張量中的TensorFlow調整值

Question

如何在TensorFlow張量中執行以下操作？

在矩陣A：如果A [I，J]> 1，則A [I，J] = 1

（在numpy的我這樣做：A[A>1] = 1）

Answer 1

可以使用tf.minimum，這按元素最小化計算;

A = tf.constant([-1, 0, 1, 3, 4]) 

A_clipped = tf.minimum(A, 1) 

sess = tf.InteractiveSession() 
A_clipped.eval() 
# array([-1, 0, 1, 1, 1], dtype=int32) 

---

另一種選擇是使用tf.where設定值：

tf.where(A > 1, tf.constant(1, shape=A.shape), A).eval() 
# array([-1, 0, 1, 1, 1], dtype=int32) 

---

如果你需要更新通過設置y = 1，在x值將與最大的1被裁剪變量A：

A = tf.Variable([-1, 0, 1, 3, 4]) 
​ 
tf.global_variables_initializer().run() 
tf.assign(A, tf.minimum(A, 1)).eval() 

A.eval() 
# array([-1, 0, 1, 1, 1], dtype=int32)