- chainer.backends.cuda.to_gpu(array, device=None, stream=None)¶
Copies the given CPU array to the specified device.
array (array, None, list or tuple) – Array or arrays to be sent to GPU.
device – CUDA device specifier. If
cuda.DummyDevice, the arrays will be copied to the current CUDA device.
stream (Stream) – (deprecated since v3.0.0) CUDA stream. If not
None, the copy runs asynchronously.
Array or arrays on GPU.
If some of the arrays are already on GPU, then this function just returns those arrays without performing any copy.
If input arrays include None, it is returned as None as is.
- Return type