tensorflow之修改数组的值

有时候需要把数组中比较小的值过滤掉，

示例：

data1 = np.array([1,2,3])
data2 = np.where(data1 > 1,data1,0)
print(data2)

输出结果

可以看出，如果data1 大于1 那么取值为本来的值，如果小于1 则为0