Python mxnet.ndarray.save用法及代碼示例

用法: mxnet.ndarray.save(fname, data)

參數：

fname：(str) - 文件名。
data：(NDArray, mxnet.ndarray.sparse.RowSparseNDArray or mxnet.ndarray.sparse.CSRNDArray, or list of NDArray, mxnet.ndarray.sparse.RowSparseNDArray or mxnet.ndarray.sparse.CSRNDArray, or dict of str to NDArray, mxnet.ndarray.sparse.RowSparseNDArray or mxnet.ndarray.sparse.CSRNDArray) - 要保存的數據。

將數組列表或 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.save。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。