fromDlpack(dltensor) → ndarray¶
Zero-copy conversion from a DLPack tensor to a
DLPack is a open in memory tensor structure proposed in this repository: dmlc/dlpack.
This function takes a
PyCapsuleobject which contains a pointer to a DLPack tensor as input, and returns a
ndarray. This function does not copy the data in the DLPack tensor but both DLPack tensor and
ndarrayhave pointers which are pointing to the same memory region for the data.
Parameters: dltensor (
PyCapsule) – Input DLPack tensor which is encapsulated in a
Returns: A CuPy ndarray. Return type: array (
>>> import cupy >>> array1 = cupy.array([0, 1, 2], dtype=cupy.float32) >>> dltensor = array1.toDlpack() >>> array2 = cupy.fromDlpack(dltensor) >>> cupy.testing.assert_array_equal(array1, array2)