输出层有四个节点，但我想使用其中一个节点的输出，我该如何解决？

Question

输出层代码：

Weights1 = tf.Variable(tf.random_normal([11, 4]))

biases1 = tf.Variable(tf.zeros([1, 4]) + 0.1)
Wx_plus_b1 = tf.matmul(l0, Weights1) + biases1

N1act = 2/(1+pow(math.e,-Wx_plus_b1[3]))-1 

我想使用第四个节点的输出

这是我自定义的激活函数，它只需要一个输入。

prediction = tf_spiky(N1act) 

错误信息：

引发 ValueError(err.message)

ValueError: 维度 0 的切片索引 3 超出范围。为了 'strided_slice' (op: 'StridedSlice') 输入形状：[1,4], [1], [1], [1] 和计算输入张量： input[1] = , input[2] = , 输入[3] = .

Answer 1

tf.matmul(l0, Weights1) 和 biases1 都具有形状 [1,4]，因此Wx_plus_b1 也是如此。也就是说，Wx_plus_b1 是一个一行四列的矩阵。当您编写 Wx_plus_b1[3] 时，您正在选择不存在的 Wx_plus_b1 的第 4 行，因此出现错误。您要查找的值实际上是Wx_plus_b1[0,3]，即第一行第四列中的值。