cupy.load¶
-
cupy.
load
(file, mmap_mode=None)[source]¶ Loads arrays or pickled objects from
.npy
,.npz
or pickled file.This function just calls
numpy.load
and then sends the arrays to the current device. NPZ file is converted to NpzFile object, which defers the transfer to the time of accessing the items.Parameters: - file (file-like object or string) – The file to read.
- mmap_mode (None, 'r+', 'r', 'w+', 'c') – If not
None
, memory-map the file to construct an intermediatenumpy.ndarray
object and transfer it to the current device.
Returns: CuPy array or NpzFile object depending on the type of the file. NpzFile object is a dictionary-like object with the context manager protocol (which enables us to use with statement on it).
See also