jax.numpy.load

目录

jax.numpy.load#

jax.numpy.load(*args, **kwargs)[源代码][源代码]#

从 .npy, .npz 或序列化文件中加载数组或序列化对象。

LAX-后端实现 numpy.load()。

原始文档字符串如下。

警告

加载包含对象数组的文件使用 pickle 模块，该模块对错误或恶意构造的数据不安全。考虑传递 allow_pickle=False 来加载已知不包含对象数组的数据，以更安全地处理不受信任的来源。

参数:

  • file (file-like object, string, or pathlib.Path) – 要读取的文件。类文件对象必须支持 seek() 和 read() 方法，并且必须始终以二进制模式打开。Pickled 文件还要求类文件对象支持 readline() 方法。

  • mmap_mode ({None, 'r+', 'r', 'w+', 'c'}, optional) – 如果不是 None，则使用给定的模式（参见 numpy.memmap 以获取模式的详细描述）对文件进行内存映射。内存映射数组保存在磁盘上。然而，它可以像任何 ndarray 一样被访问和切片。内存映射对于访问大文件中的小片段而不将整个文件读入内存特别有用。

  • allow_pickle (bool, optional) – 允许加载存储在npy文件中的pickle对象数组。不允许pickle的原因包括安全性，因为加载pickle数据可以执行任意代码。如果禁止pickle，加载对象数组将会失败。默认值：False .. versionchanged:: 1.16.3 在响应CVE-2019-6446后，默认值改为False。

  • fix_imports (bool, optional) – 仅在Python 3上加载由Python 2生成的pickle文件时才有用，这包括包含对象数组的npy/npz文件。如果`fix_imports`为True，pickle将尝试将旧的Python 2名称映射到Python 3中使用的新名称。

  • encoding (str, optional) – 读取Python 2字符串时使用的编码。仅在Python 3中加载由Python 2生成的pickle文件时才有用，这包括包含对象数组的npy/npz文件。除了’latin1’、’ASCII’和’bytes’之外的值是不允许的，因为它们可能会损坏数值数据。默认值：’ASCII’

  • max_header_size (int, optional) – 允许的最大头部大小。较大的头部可能不安全加载，因此需要显式传递一个更大的值。详情请参见 ast.literal_eval()。当 allow_pickle 被传递时，此选项将被忽略。在这种情况下，文件被认为是可信的，因此限制是不必要的。

  • args (Any)

  • kwargs (Any)

返回:

结果 – 文件中存储的数据。对于 .npz 文件，返回的 NpzFile 类实例必须关闭以避免文件描述符泄漏。

返回类型:

array, tuple, dict, etc.

目录