jax.numpy.from_dlpack
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
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