0
我知道tf.nn.dynamic_rnn
通過將最後一個有效狀態複製到輸出向量的末尾來處理可變長度,並用零向量填充LSTM輸出。TensorFlow動態RNN - 如何對所有LSTM狀態取平均值而忽略零矢量?
在這種情況下,我想對所有LSTM狀態做一個平均值,忽略零向量。我如何在TensorFlow中做到這一點?
或者,你如何在掩蔽零的同時在矢量列表上取平均值?
謝謝!
我知道tf.nn.dynamic_rnn
通過將最後一個有效狀態複製到輸出向量的末尾來處理可變長度,並用零向量填充LSTM輸出。TensorFlow動態RNN - 如何對所有LSTM狀態取平均值而忽略零矢量?
在這種情況下,我想對所有LSTM狀態做一個平均值,忽略零向量。我如何在TensorFlow中做到這一點?
或者,你如何在掩蔽零的同時在矢量列表上取平均值?
謝謝!
Calcuate的總和,然後通過非零條目,您可以使用tf.count_nonzero
做一個總和,而不是平均,然後由長度除以獲得的平均計數的數量劃分。 – Aaron