如何在使用tf.nn.softmax計算softmax時避免填充索引？

Question

我有一個矩陣，每行填充maxlen。如果我在矩陣上應用softmax，也會考慮填充指數。如何在僅考慮非填充指數的情況下將softmax應用於矩陣？

注意：矩陣每一行中的填充長度會有所不同。

Answer 1

您可以將填充的單元格設置爲接近-Infinity的值（即與您的其他邏輯相比非常小），以便它們的貢獻可以忽略不計。