如何在TensorFlow中清除Tensor中的条目?

时间:2016-09-20 15:35:57

标签: tensorflow

经过一些计算后,我获得了Tensor x 的形状(128,128,32)。 我怎样才能实现与下一个代码相同的行为?

x[:, :, 1:] = 0

实际上,我想清除除一个矩阵之外的所有条目。

张量是在会话中获得的,我想清除同一会话中的条目。

最简单的方法是什么?

1 个答案:

答案 0 :(得分:1)

从0.10开始,TensorFlow没有可用于此目的的高级索引,因此您需要重新构造为矢量,分配给线性索引集,然后重新整形。您可以使用变量+ scatter_update或dynamic_stitch指定here

对矢量执行此分配