numpy.from_dlpack #
- 麻木的。from_dlpack ( x , / ) #
从实现协议的对象创建 NumPy 数组
__dlpack__
。通常,返回的 NumPy 数组是输入对象的只读视图。有关更多详细信息,请参阅[1]和[2] 。- 参数:
- x对象
__dlpack__
实现和 方法的Python 对象__dlpack_device__
。
- 返回:
- 输出数组
参考
[ 1 ][ 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)