chainer.cuda.get_device_from_array¶
-
chainer.cuda.
get_device_from_array
(*arrays)[source]¶ Gets the device from a list of CuPy array or a single CuPy array.
The device on which the given CuPy array reside is returned.
Parameters: array (cupy.ndarray or list of cupy.ndarray) – A CuPy array which this function returns the device corresponding to. If a list of cupy.ndarray
s are given, it returns the first device object of an array in the list.