经过一些计算后,我获得了Tensor x 的形状(128,128,32)。 我怎样才能实现与下一个代码相同的行为?
x[:, :, 1:] = 0
实际上,我想清除除一个矩阵之外的所有条目。
张量是在会话中获得的,我想清除同一会话中的条目。
最简单的方法是什么?
答案 0 :(得分:1)
从0.10开始,TensorFlow没有可用于此目的的高级索引,因此您需要重新构造为矢量,分配给线性索引集,然后重新整形。您可以使用变量+ scatter_update或dynamic_stitch指定here
对矢量执行此分配