TensorFlow动态RNN - 如何取消忽略零向量的所有LSTM状态的均值？

Question

我理解tf.nn.dynamic_rnn通过将最后一个有效状态复制到输出向量的末尾来处理变量长度，并用零向量填充LSTM输出。

在这种情况下，我想忽略零向量，对所有LSTM状态做一个均值。我怎么能在TensorFlow中做到这一点？

或者，如何在屏蔽零的同时对一个向量列表取平均值？

谢谢！

Answer 1

计算总和，然后除以非零项的数量，您可以使用tf.count_nonzero计算其数