昇腾社区首页
中文
注册

multi_save接口

接口功能

将同一个数据保存到多个文件中。

接口格式

mindio_acp.multi_save(obj, path_list)

接口参数

参数

是否必选

说明

取值要求

obj

必选

需要保存的对象。

有效数据对象。

path_list

必选

数据保存路径列表。

有效文件路径列表。

使用样例

>>> # Save to file
>>> x = torch.tensor([0, 1, 2, 3, 4])
>>> path_list = ["/mnt/dpc01/dir1/rank_1.pt","/mnt/dpc01/dir2/rank_1.pt"]
>>> mindio_acp.multi_save(x, path_list)

返回值

  • None:失败。
  • 0:使用原生torch.save方式保存成功。
  • 1:使用memfs方式保存成功。
  • 2:使用fopen保存成功。