Shortcuts

torch.utils.dlpack

torch.utils.dlpack.from_dlpack(ext_tensor)Tensor[source]

Convers a tensor from a external library into a torch.Tensor by means of the __dlpack__ protocol.

The tensor will share the memory with the object represented in the DLPack.

Warning

Only call from_dlpack once per capsule. Its behavior when used on the same capsule multiple times is undefined.

Parameters

ext_tensor (object with __dlpack__ attribute or DLPack capsule) – The tensor or DLPack capsule to convert.

torch.utils.dlpack.to_dlpack(tensor)PyCapsule

Returns a DLPack representing the tensor.

Parameters

tensor – a tensor to be exported

The DLPack shares the tensors memory. Note that each DLPack can only be consumed once.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources