文档
注册

save接口

接口功能

将数据保存到指定的UnderFS路径下。

接口格式

torch_mindio.save(obj, path, open_way='memfs')

接口参数

参数

是否必选

说明

取值要求

obj

必选

需要保存的对象。

有效数据对象。

path

必选

数据保存路径。

有效UnderFS路径。

open_way

可选

保存方式,memfs是使用MindIO的高性能MemFS保存数据;fopen是调用C标准库中的文件操作函数保存数据,通常作为memfs方式的备份存在。默认值:memfs。

  • memfs
  • fopen

使用样例

>>> # Save to file
>>> x = torch.tensor([0, 1, 2, 3, 4])
>>> torch_mindio.save(x, '/mnt/dpc01/tensor.pt')

返回值

None

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词