tensorflow：逐块数组/矩阵和

Question

是否有任何张量流函数允许人们对数组或矩阵进行逐块求和。

例如，如果我有一个数组

tf.constant(np.array([1,2,3,4,5,6]))

我需要返回一组2元素块的数组。换句话说，数组中1 + 2, 3 + 4和5 + 6的总和。

np.array([3,7,11])

Answer 1

您可以使用tf.reshape和tf.reduce_sum执行此操作，如下所示：

import tensorflow as tf
n=tf.constant([1,2,3,4,5,6])
res=tf.reduce_sum(tf.reshape(n,(3,2)),1)
sess=tf.Session()
res.eval(session=sess)

这会打印array([ 3, 7, 11])

在一般情况下，如果您想获得k元素的不相交序列的总和，请将tf.reshape(n,(3,2))替换为tf.reshape(n,(-1,k))。 （-1将强制它指定n/k的行数。）