我正在使用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)
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句