日韩av黄I国产麻豆传媒I国产91av视频在线观看I日韩一区二区三区在线看I美女国产在线I麻豆视频国产在线观看I成人黄色短片

歡迎訪問 生活随笔!

生活随笔

當(dāng)前位置: 首頁 >

如何保存Tensorflow中的Tensor参数,保存训练中的中间参数,存储卷积层的数据

發(fā)布時(shí)間:2025/4/16 62 豆豆
生活随笔 收集整理的這篇文章主要介紹了 如何保存Tensorflow中的Tensor参数,保存训练中的中间参数,存储卷积层的数据 小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.

在自己構(gòu)建的卷積神經(jīng)時(shí),我想把卷積層的數(shù)據(jù)提取出來,但是這些數(shù)據(jù)是Tensor類型的

網(wǎng)上幾乎找不到怎么存儲(chǔ)的例子,然后被我發(fā)下了一下解決辦法?

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

import tensorflow as tf #輸入為100個(gè)數(shù)據(jù)集,28*28像素,3個(gè)通道 input = tf.Variable(tf.random_normal([100,28,28,3])) ref = tf.Variable(tf.zeros([100,14,14,3]),tf.float32)#滑動(dòng)窗口的尺寸為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

?

轉(zhuǎn)載于:https://www.cnblogs.com/imzh/p/10153358.html

總結(jié)

以上是生活随笔為你收集整理的如何保存Tensorflow中的Tensor参数,保存训练中的中间参数,存储卷积层的数据的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò),歡迎將生活随笔推薦給好友。