当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python mxnet.ndarray.utils.save用法及代码示例


用法:

mxnet.ndarray.utils.save(fname, data)

参数

将数组列表或 str->array 的字典保存到文件中。

文件名示例:

  • /path/to/file
  • s3://my-bucket/path/to/file(如果使用 AWS S3 支持编译)
  • hdfs://path/to/file(如果编译时支持 HDFS)

例子

>>> x = mx.nd.zeros((2,3))
>>> y = mx.nd.ones((1,4))
>>> mx.nd.save('my_list', [x,y])
>>> mx.nd.save('my_dict', {'x':x, 'y':y})
>>> mx.nd.load('my_list')
[<NDArray 2x3 @cpu(0)>, <NDArray 1x4 @cpu(0)>]
>>> mx.nd.load('my_dict')
{'y': <NDArray 1x4 @cpu(0)>, 'x': <NDArray 2x3 @cpu(0)>}

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.utils.save。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。