1. 程式人生 > >如何保存Tensorflow中的Tensor參數,保存訓練中的中間參數,存儲卷積層的數據

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

put pool 數據 random ack 滑動 orm over 尺寸

在自己構建的卷積神經時,我想把卷積層的數據提取出來,但是這些數據是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

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