tensorflow怎么获得某一层的输出值

更新时间:2023-06-09 10:43:15 阅读: 评论:0

tensorflow怎么获得某⼀层的输出值tensorflow=2.0+
在使⽤tensorflow加载模型的时候有时候需要查看这个模型某⼀层的输出。
搭建⼀个简单的神经⽹络,识别cifar数据集:
点击查看代码
model = dels.Sequential()金莎
跨越中国制造
升级考核model.add(Flatten())
model.add(BatchNormalization())
model.add(Activation('relu'))
吹风机多大功率的好
model.add(Den(1024, activation='relu', name="den_test1"))深圳市龙岗区
model.add(Den(2048, activation='relu', name="den_test2"))
model.add(Den(2048, activation='relu', name="den_test3"))
model.add(Den(2048, activation='relu', name="den_test4"))
形容相同的成语
model.add(Den(2048, activation='relu', name="den_test5"))
湖南特色美食小吃
model.add(Den(10, activation='softmax', name="den_test6"))
搭建好后进⾏训练可以得到⼀个训练好的神经⽹络
此时通过模型去识别cifar测试集
model.predict(x_test)#x_test是cifar数据集
可以得到预测结果,此时若想查看中间某⼀层的输出,⽐如全连接的第⼆层——den_test2的输出怎么办呢。这时候,可以直接截取此模型的⼦模型,直接将 den_test2作为最后⼀层输出。
sub_model = tf.keras.Model(inputs = model.input, outputs = _layer(den_test2).output)
淀粉跟面粉的区别
sub_model.predict(x_test)
这样就可以获取中间某⼀层的输出了。

本文发布于:2023-06-09 10:43:15,感谢您对本站的认可!

本文链接:https://www.wtabcd.cn/fanwen/fan/89/1030499.html

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。

标签:模型   查看   训练   得到   识别   时候
相关文章
留言与评论(共有 0 条评论)
   
验证码:
推荐文章
排行榜
Copyright ©2019-2022 Comsenz Inc.Powered by © 专利检索| 网站地图