2016-12-05 41 views
0

我知道tf.nn.dynamic_rnn通過將最後一個有效狀態複製到輸出向量的末尾來處理可變長度,並用零向量填充LSTM輸出。TensorFlow動態RNN - 如何對所有LSTM狀態取平均值而忽略零矢量?

在這種情況下,我想對所有LSTM狀態做一個平均值,忽略零向量。我如何在TensorFlow中做到這一點?

或者,你如何在掩蔽零的同時在矢量列表上取平均值?

謝謝!

+0

做一個總和,而不是平均,然後由長度除以獲得的平均計數的數量劃分。 – Aaron

回答