更新张量中的切片或列

Question

无论张量的形状如何，我都试图更新整个列，而且列的索引也可以是形状范围内的任何值。

pk

我正在尝试实现此numpy操作：

 tensor = tf.Variable(tf.ones((5,5)))

我尝试使用tf.scatter_update，但没有运气。

Answer 1

如果您不介意创建新的张量，可以尝试：

import tensorflow as tf

t = tf.Variable(tf.ones((5,5)))
s0, s1, s2 = tf.split(t, [2, 1, 2], axis=1)
s = tf.concat([s0,tf.zeros((5,1)),s2], axis=1)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    y = sess.run(s)
    print(y)