numpy.from_dlpack#

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

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

Параметры:

xobject: Объект Python, который реализует __dlpack__ и __dlpack_device__ методы.
deviceустройство, опционально: Устройство, на котором размещается созданный массив. По умолчанию: None. Должно быть "cpu" если передано, что может позволить импортировать массив, который ещё не доступен на CPU.
copybool, необязательно: Логическое значение, указывающее, копировать ли входные данные. Если True, будет создана копия. Если False, функция никогда не будет копировать и вызовет BufferError в случае, если копия считается необходимой. Передача этого запрашивает копию у экспортера, который может или не может реализовать эту возможность. Если None, функция повторно использует существующий буфер памяти, если это возможно, и копирует в противном случае. По умолчанию: None.

Возвращает:

выходndarray

Ссылки

[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)  