1. 程式人生 > >如何儲存Tensorflow中的Tensor引數,儲存訓練中的中間引數,儲存卷積層的資料

如何儲存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