numpy.from_dlpack#
- numpy.from_dlpack(x, /, *, device=None, copy=None)#
从实现了
__dlpack__协议的对象创建 NumPy 数组。通常，返回的 NumPy 数组是对输入对象的视图。有关更多详细信息，请参阅 [1] 和 [2]。- 参数:
- xobject
实现了
__dlpack__和__dlpack_device__方法的 Python 对象。- devicedevice, optional
放置创建的数组的设备。默认值：
None。如果传递，则必须是"cpu"，这可能允许导入一个尚未可用的 CPU 数组。- copybool, optional
一个布尔值，指示是否复制输入。如果为
True，则会进行复制。如果为False，则该函数绝不会复制，并且在需要复制时会引发BufferError。传递它会请求导出器进行复制，而导出器可能实现也可能不实现此功能。如果为None，则函数将在可能的情况下重用现有的内存缓冲区，否则进行复制。默认值：None。
- 返回:
- outndarray
参考
[1]Array API 文档，https://data-apis.org/array-api/latest/design_topics/data_interchange.html#syntax-for-data-interchange-with-dlpack
[2]DLPack 的 Python 规范，https://dmlc.github.io/dlpack/latest/python_spec.html
示例
>>> import torch >>> x = torch.arange(10) >>> # create a view of the torch tensor "x" in NumPy >>> y = np.from_dlpack(x)