我想复制以下numpy代码tensorflow.例如,我想为0之前具有值的所有张量索引分配一个1.
a = np.array([1, 2, 3, 1])
a[a==1] = 0
# a should be [0, 2, 3, 0]
如果我写相似的代码tensorflow我得到以下错误.
TypeError: 'Tensor' object does not support item assignment
方括号中的条件应该是任意的,如a[a<1] = 0.
有没有办法实现这种"条件分配"(缺少一个更好的名字)tensorflow？