如何將函數應用於張量張量的給定列？

Question

說我有

v1 = tf.Variable(tf.random_uniform(shape=[2,3])) 

，我想這樣做

v1[:,1] = tf.log(1+tf.exp(v1[:,1])) 

這不TensorFlow工作。

如何將函數轉換/應用於列？

Answer 1

您可以使用tf.slice作爲來自張量的選取列，並且可以使用tf.scatter_nd_update將其歸零。