如何保存Tensorflow中的Tensor參數,保存訓練中的中間參數,存儲卷積層的數據


在自己構建的卷積神經時,我想把卷積層的數據提取出來,但是這些數據是Tensor類型的

網上幾乎找不到怎么存儲的例子,然后被我發下了一下解決辦法 

https://stackoverflow.com/questions/41587689/how-to-save-a-tensor-in-checkpoint-in-tensorflow

import tensorflow as tf
#輸入為100個數據集,28*28像素,3個通道
input = tf.Variable(tf.random_normal([100,28,28,3]))
ref   = tf.Variable(tf.zeros([100,14,14,3]),tf.float32)

#滑動窗口的尺寸為2*2,步幅為2
pool = tf.nn.max_pool(input, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding="SAME")
pool = tf.assign (ref ,pool)
pool

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM