cupy.load#

cupy.load(file, mmap_mode=None, allow_pickle=None)[源代码][源代码]#

从 .npy, .npz 或 pickle 文件加载数组或 pickled 对象。

这个函数只是调用 numpy.load 然后将数组发送到当前设备。NPZ 文件被转换为 NpzFile 对象，该对象将传输推迟到访问项目时。

参数:

  • file (file-like object or string) – 要读取的文件。

  • mmap_mode (None, 'r+', 'r', 'w+', 'c') – 如果不是 None ，则将文件映射到内存中以构建一个中间的 numpy.ndarray 对象，并将其传输到当前设备。

  • allow_pickle (bool) – 允许加载存储在npy文件中的pickled对象数组。不允许pickles的原因包括安全性，因为加载pickled数据可以执行任意代码。如果不允许pickles，加载对象数组将会失败。请注意，CuPy不支持dtype为`object`的数组。默认值为False。此选项仅适用于NumPy 1.10或更高版本。在NumPy 1.9中，无法指定此选项（始终允许加载pickled对象）。

返回:

根据文件类型，可能是 CuPy 数组或 NpzFile 对象。NpzFile 对象是一个类似字典的对象，具有上下文管理器协议（这使得我们可以在其上使用 with 语句）。

参见

numpy.load()