save接口
接口功能
将数据保存到指定的UnderFS路径下。
接口格式
torch_mindio.save(obj, path, open_way='memfs')
接口参数
参数 |
是否必选 |
说明 |
取值要求 |
---|---|---|---|
obj |
必选 |
需要保存的对象。 |
有效数据对象。 |
path |
必选 |
数据保存路径。 |
有效UnderFS路径。 |
open_way |
可选 |
保存方式,memfs是使用MindIO的高性能MemFS保存数据;fopen是调用C标准库中的文件操作函数保存数据,通常作为memfs方式的备份存在。默认值:memfs。 |
|
使用样例
>>> # Save to file >>> x = torch.tensor([0, 1, 2, 3, 4]) >>> torch_mindio.save(x, '/mnt/dpc01/tensor.pt')
返回值
None
父主题: API接口参考