python - 在评估 TensorFlow 模型时,如何在单个前向调用中查看特定隐藏层的值?
问题描述
我定义了一个简单的顺序模型如下:
m = tf.keras.models.Sequential([
tf.keras.layers.Dense(10, input_shape=(3,), name='fc1', activation='relu'),
tf.keras.layers.Dense(3, input_shape=(3,), name='fc2'),
])
m.summary()
Model: "sequential_6"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
fc1 (Dense) (None, 10) 40
_________________________________________________________________
fc2 (Dense) (None, 3) 33
=================================================================
Total params: 73
Trainable params: 73
Non-trainable params: 0
_________________________________________________________________
现在,我得到一个隐藏层layer_fc1
,如下所示:
layer_fc1 = m.get_layer('fc1')
layer_fc1
<tensorflow.python.keras.layers.core.Dense at 0x7f6fcc7d9eb8>
现在,我想看看当我评估这个模型时,我想看看 layer 的值fc1
。
整个网络和layer_fc1
单独的前向调用的评估如下:
t = tf.random.uniform((1, 3))
m(t)
<tf.Tensor: id=446892, shape=(1, 3), dtype=float32, numpy=array([[ 0.0168661 , -0.12582873, -0.0308626 ]], dtype=float32)>
layer_fc1(t)
<tf.Tensor: id=446904, shape=(1, 10), dtype=float32, numpy=
array([[0. , 0. , 0.00877494, 0.05680769, 0.08027849,
0.12362152, 0. , 0. , 0.22683921, 0.17538759]],
dtype=float32)>
那么,有没有什么方法可以通过单次转发调用m(t)
,我也得到该隐藏层的值?这样,计算将更有效率。
解决方案
您可以使用功能API 轻松完成此操作,如下所示:
inpt = tf.keras.layers.Input(shape=(3,))
fc1_out = tf.keras.layers.Dense(10, name='fc1', activation='relu')(inpt)
fc2_out = tf.keras.layers.Dense(3, name='fc2')(fc1_out)
m = tf.keras.models.Model(inputs=inpt, outputs=[fc2_out, fc1_out])
t = tf.random.uniform((1,3))
m(t)
这给出了您正在寻找的输出:
[<tf.Tensor: id=149, shape=(1, 3), dtype=float32, numpy=array([[-0.20491418, -0.33935153, 0.07777037]], dtype=float32)>,
<tf.Tensor: id=145, shape=(1, 10), dtype=float32, numpy=
array([[0. , 0.5071918 , 0. , 0.24756521, 0.05489198,
0.31997102, 0. , 0.23788954, 0.01050918, 0.24083027]],
dtype=float32)>]
我对 Sequential API 不太熟悉,但我希望这在 Sequential API 中是不可能的,因为对我来说,这不是一个顺序模型,其中一层从输入到输出跟随另一层。
推荐阅读
- c# - 如何按月对数据表进行分组并计算该月以下的行数
- json - 读取外部 JSON 文件
- android - 仅在没有缺口的手机上录制和工作时,媒体记录器变黑
- printthis - printThis - 不要求选择打印机
- node.js - 节点获取:[DEP0066] DeprecationWarning:OutgoingMessage.prototype._headers 已弃用
- angular - 离子输入 e2e 测试导致元素不可交互
- sql - Oracle SQL:显示按列分组的行的行号
- angular - 上传 PUT 请求由 Angular ServiceWorker 发送两次
- security - Elixir 有源代码分析工具吗?
- html - 为什么 div 文本不在块周围浮动?