在张量流中，如何将张量的行重置为零？

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)