numpy.from_dlpack

numpy.from_dlpack(x, /)

Create a NumPy array from an object implementing the __dlpack__ protocol. Generally, the returned NumPy array is a read-only view of the input object. See [1] and [2] for more details.

Parameters:

xobject

A Python object that implements the __dlpack__ and __dlpack_device__ methods.

Returns:

outndarray

References

[1]

Array API documentation, https://data-apis.org/array-api/latest/design_topics/data_interchange.html#syntax-for-data-interchange-with-dlpack

[2]

Python specification for DLPack, https://dmlc.github.io/dlpack/latest/python_spec.html

Examples

>>> import torch
>>> x = torch.arange(10)
>>> # create a view of the torch tensor "x" in NumPy
>>> y = np.from_dlpack(x)