numpy.ndarray.to_device

方法

ndarray.to_device(device, /, *, stream=None)

为兼容 Array API 而设。由于 NumPy 仅支持 CPU 数组，此方法是一个无操作，返回相同的数组。

参数:

device“cpu”

必须是 "cpu"。

streamNone, optional

目前不支持。

返回:

outSelf

返回相同的数组。