multi_save接口

接口功能

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

接口格式

torch_mindio.multi_save(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"]
>>> torch_mindio.multi_save(x, path_list)

返回值

• None：失败。
• 0：成功。