numpy.from_dlpack#

numpy.from_dlpack(x, /, *, device=None, copy=None)#

从实现 __dlpack__ 协议的对象创建 NumPy 数组。通常，返回的 NumPy 数组是输入对象的视图。更多详情请参阅 [1] 和 [2]。

参数:
x对象

一个实现了 __dlpack__ 和 __dlpack_device__ 方法的 Python 对象。

device设备，可选

用于放置所创建数组的设备。默认值: None。如果传入，则必须是 "cpu"，这可能允许导入尚未在 CPU 上可用的数组。

copy布尔值，可选

指示是否复制输入的布尔值。如果为 True，则会进行复制。如果为 False，函数将永远不复制，并且在需要复制时会引发 BufferError。传递此参数会向导出者请求复制，导出者可能实现也可能未实现此功能。如果为 None，函数将尽可能重用现有内存缓冲区，否则进行复制。默认值: None。

返回:
outndarray

参考

示例

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