在張量流中，如何將張量的行重置爲零？

Question

我正在研究Tensorflow中的遞歸神經網絡。 我有張量H它存儲一批輸入的隱藏狀態。

假設H的大小爲a*b。如何將列表H的行重置爲zeros？

例如如果列表是[1, 2]我想只復位H到zeros的那些行，使其他的完好無損。

Answer 1

您確定要使用張量張量來做這個嗎？我不確定你的確切用例是什麼，但我想不出你需要這樣做的原因。你應該能夠將你需要的東西傳遞給一個tf.placeholder。

無論如何，要回答你的問題，你可以乘以你想要重置張量的常數張量與1的每一個除了你需要重置的地方。

例如：

reset_constant = tf.constant([[1,1,1], 
           [1,1,1], 
           [0,0,0], #reset this row 
           [1,1,1]], dtype="float") 
tensor_with_reset_row = tf.mul(tensor_to_reset, reset_constant)