神经网络-在张量流中，如何将一行张量重置为零

Question

我正在Tensorflow中研究递归神经网络。我有一个张量H，它为一批输入存储隐藏状态。

假设CCD_ 2的大小为CCD_。如何将列表中给定的H行重置为zeros？

例如，如果列表是[1, 2]，则我只想将H的那些行重置为zeros，而保留其他行不变。

Answer 1

您确定要使用tensorflow张量执行此操作吗？我不确定您的确切用例是什么，但我想不出您需要这样做的原因。你应该能够将你需要的东西准确地传递到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)