2016-07-29 13 views
0

私はTensorflowのリカレントニューラルネットワークに取り組んでいます。 テンソルHは、入力バッチの隠れた状態を保存しています。テンソルフローでは、テンソルの行をゼロにリセットするにはどうすればよいですか?

Hは、サイズがa*bであるとします。リスト内のHの行をzerosにリセットするにはどうすればよいですか?

リストが[1, 2]の場合Hの行だけをzerosにリセットし、他の行はそのまま残したいと考えています。

答えて

0

テンソルフローテンソルを使用してこれを実行してもよろしいですか?私はあなたの正確なユースケースが何であるかはわかりませんが、あなたがこれを行う必要がある理由を考えることはできません。必要なものをtf.placeholderに渡すことができるはずです。

とにかく、あなたの質問に答えるために、リセットする必要のあるテンソル以外に、テンソルを定数で置き換えることができます。例えば

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) 
関連する問題