2016-07-29 257 views
0

我正在研究Tensorflow中的遞歸神經網絡。 我有張量H它存儲一批輸入的隱藏狀態。在張量流中,如何將張量的行重置爲零?

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

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

回答

0

您確定要使用張量張量來做這個嗎?我不確定你的確切用例是什麼,但我想不出你需要這樣做的原因。你應該能夠將你需要的東西傳遞給一個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) 
相關問題