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

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将其放回。