在张量流中,如何将张量的行重置为零?
问题描述:
我正在研究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)