python中的张量切片中的值总和

Question

我正在使用python / tensorflow。我有一个张量x（4个维度），并希望将总和覆盖张量的一部分

np.sum(x[:,:,:,0])

我得到了错误

ValueError: setting an array element with a sequence.

我在做什么错了？

Answer 1

假设您正在使用TF 1.x，则无法对TF张量（在TF 1.x中）执行简单的numpy操作。而是执行以下操作。

import tensorflow as tf
x = tf.random_normal(shape=[10, 5, 3, 2])
res = tf.reduce_sum(tf.gather(x, 0, axis=-1))

with tf.Session() as sess:
  r = sess.run(res)