numpy.from_dlpack

numpy.from_dlpack(x, /)

Создать массив NumPy из объекта, реализующего протокол __dlpack__. Обычно возвращаемый массив NumPy является только чтением видом входного объекта. Подробнее см. [1] и [2].

Параметры:

xобъект

Объект Python, реализующий методы __dlpack__ и __dlpack_device__.

Возвращаемое значение:

outndarray

Ссылки

[1]

Документация Array API, https://data-apis.org/array-api/latest/design_topics/data_interchange.html#syntax-for-data-interchange-with-dlpack

[2]

Спецификация Python для DLPack, 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)