numpy.from_dlpack

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

從實作 __dlpack__ 協定的物件建立 NumPy 陣列。通常，返回的 NumPy 陣列是輸入物件的唯讀視圖。詳情請參閱 [1] 和 [2]。

參數:

x物件

一個 Python 物件，實作了 __dlpack__ 和 __dlpack_device__ 方法。

devicedevice，選用

放置建立陣列的裝置。預設值：None。如果傳遞，則必須為 "cpu"，這可能允許匯入尚未在 CPU 上可用的陣列。

copybool，選用

布林值，指示是否複製輸入。如果為 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)