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,(-1,k))代替tf.reshape(n,(3,2))。 （-1會強制它指定行數爲n/k）。