jax.numpy.from_dlpack#

jax.numpy.from_dlpack(x)[source]#

Create a NumPy array from an object implementing the __dlpack__

LAX-backend implementation of numpy.from_dlpack().

Original docstring below.

protocol. Generally, the returned NumPy array is a read-only view of the input object. See 1 and 2 for more details.

Parameters

x (object) – A Python object that implements the __dlpack__ and __dlpack_device__ methods.

Returns

out

Return type

ndarray

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