View page source

warp.from_torch#

warp.from_torch(
t,
dtype=None,
requires_grad=None,
grad=None,
return_ctype=False,
retain_grad=False,
)[source]#

Convert a Torch tensor to a Warp array without copying the data.

Parameters:

  • t (torch.Tensor) – The torch tensor to wrap.

  • dtype (type | None) – The target data type of the resulting Warp array. Defaults to the tensor value type mapped to a Warp array value type.

  • requires_grad (bool | None) – Whether the resulting array should wrap the tensor’s gradient, if it exists (the grad tensor will be allocated otherwise). Defaults to the tensor’s requires_grad value.

  • grad – Optional gradient array to attach to the result. Can be a Warp array or Torch tensor. If not provided and requires_grad is True, the tensor’s gradient will be wrapped or allocated.

  • return_ctype (bool) – Whether to return a low-level array descriptor instead of a warp.array object (faster). The descriptor can be passed to Warp kernels.

  • retain_grad (bool) – Whether to preserve gradients during backward instead of zeroing after read.

Returns:

The wrapped array or array descriptor.

Return type:

warp.array | warp._src.types.array_t