mlx.core.savez

savez(file: object, *args, **kwargs) → None

将多个数组保存到未压缩的 .npz 格式二进制文件。

import mlx.core as mx

x = mx.ones((10, 10))
mx.savez("my_path.npz", x=x)

import mlx.nn as nn
from mlx.utils import tree_flatten

model = nn.TransformerEncoder(6, 128, 4)
flat_params = tree_flatten(model.parameters())
mx.savez("model.npz", **dict(flat_params))

参数:

• file (file, str) – 保存数组的文件路径。

• *args (arrays) – 要保存的数组。

• **kwargs (arrays) – 要保存的数组。每个数组将使用关联的关键字作为输出文件名保存。