mxnet基础到提高(10)--读写文件
生活随笔
收集整理的這篇文章主要介紹了
mxnet基础到提高(10)--读写文件
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
# -*- coding: utf-8 -*-
"""
Spyder EditorThis is a temporary script file.
"""
import mxnet as mx
import numpy as npimport pickle as pkl#1.pickle
a = mx.nd.ones((2, 3))
# 序列化存儲
data = pkl.dumps(a)
pkl.dump(data, open('tmp.pickle', 'wb'))
# 序列化讀取
data = pkl.load(open('tmp.pickle', 'rb'))
b = pkl.loads(data)
print b.asnumpy()
#2.直接讀寫
a = mx.nd.ones((2,3))
b = mx.nd.ones((5,6))
mx.nd.save("temp.ndarray", [a,b])
c = mx.nd.load("temp.ndarray")
print cd = {'a':a, 'b':b}
mx.nd.save("temp.ndarray", d)
c = mx.nd.load("temp.ndarray")
print c
[[1. 1. 1.]
[1. 1. 1.]]
[
[[1. 1. 1.]
[1. 1. 1.]]
<NDArray 2x3 @cpu(0)>,
[[1. 1. 1. 1. 1. 1.]
[1. 1. 1. 1. 1. 1.]
[1. 1. 1. 1. 1. 1.]
[1. 1. 1. 1. 1. 1.]
[1. 1. 1. 1. 1. 1.]]
<NDArray 5x6 @cpu(0)>]
{‘a’:
[[1. 1. 1.]
[1. 1. 1.]]
<NDArray 2x3 @cpu(0)>, ‘b’:
[[1. 1. 1. 1. 1. 1.]
[1. 1. 1. 1. 1. 1.]
[1. 1. 1. 1. 1. 1.]
[1. 1. 1. 1. 1. 1.]
[1. 1. 1. 1. 1. 1.]]
<NDArray 5x6 @cpu(0)>}
總結
以上是生活随笔為你收集整理的mxnet基础到提高(10)--读写文件的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: C指针原理(7)-C内嵌汇编
- 下一篇: mxnet基础到提高(11)--循环