cupy.from_dlpack

cupy.from_dlpack(array, *, device=None, copy=None)

符合 DLPack 数据交换协议的数组对象之间的零拷贝转换。

参数:

• array (对象) – 实现 __dlpack__() 和 __dlpack_device__() 这两个方法的数组对象。

• device (元组) – 表示 dlpack 设备的 (device_type, device_id) 元组。

• copy (布尔值|None) – 请求导出时永不或总是创建副本。默认情况下 (None)，导出器可能会或可能不会创建副本。

返回值:

一个可以在 CuPy 当前流上安全访问的 CuPy 数组。

返回类型:

cupy.ndarray

注意

此函数与 CuPy 的旧版 fromDlpack() 函数不同。此函数接受任何实现 DLPack 数据交换协议的对象作为输入，以及一个包含 DLPack 张量的原始 PyCapsule 对象（为了向后兼容性），而 fromDlpack() 只接受 PyCapsule 对象。如果输入对象不符合协议，用户有责任确保数据安全。

另请参阅

numpy.from_dlpack(), DLPack 的 Python 规范, 数据交换机制