cupy.load#

cupy.load(file, mmap_mode=None, allow_pickle=None)[source]#

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

此函数仅调用 numpy.load，然后将数组发送到当前设备。NPZ 文件被转换为 NpzFile 对象，该对象将传输延迟到访问项目时进行。

参数:

  • file (文件类对象 或 字符串) – 要读取的文件。

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

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

返回:

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

另请参阅

numpy.load()