File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed
Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -103,7 +103,7 @@ def torch_tensor_to_jax_array(torch_tensor: torch.Tensor) -> jax.Array:
103103 # https://github.com/data-apis/array-api-compat/issues/173#issuecomment-2272192054
104104 # so we explicitly resolve any conjugacy operations implied by bit before conversion
105105 torch_tensor = torch_tensor .resolve_conj ()
106- # DLPack compatibility does support tensors that require gradient so detach. As
106+ # DLPack compatibility does not support tensors that require gradient so detach. As
107107 # this intended for use when wrapping JAX code detaching tensor from gradient values
108108 # should not be problematic as derivatives will be separately routed via JAX
109109 torch_tensor = torch_tensor .detach ()
You can’t perform that action at this time.
0 commit comments