0
我正在研究Tensorflow中的遞歸神經網絡。 我有張量H
它存儲一批輸入的隱藏狀態。在張量流中,如何將張量的行重置爲零?
假設H
的大小爲a*b
。如何將列表H
的行重置爲zeros
?
例如如果列表是[1, 2]
我想只復位H
到zeros
的那些行,使其他的完好無損。
我正在研究Tensorflow中的遞歸神經網絡。 我有張量H
它存儲一批輸入的隱藏狀態。在張量流中,如何將張量的行重置爲零?
假設H
的大小爲a*b
。如何將列表H
的行重置爲zeros
?
例如如果列表是[1, 2]
我想只復位H
到zeros
的那些行,使其他的完好無損。
您確定要使用張量張量來做這個嗎?我不確定你的確切用例是什麼,但我想不出你需要這樣做的原因。你應該能夠將你需要的東西傳遞給一個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)