如何将函数应用于张量张量的给定列？

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将其归零。