reduce_sum()中的reduction_indices

参考知乎这篇答案

https://www.zhihu.com/question/51325408/answer/125426642

reduction_indices，直译过来就是指“坍塌维度”，即按照哪个维度进行加法运算。

如：

由于square(y-x)=[1,4,9,16]，是个一维向量（张量），所以reduction_indices只能设为0，设其他会报错。运算结果是30，也就是1维坍塌为0维。

将xy换成二维张量试试：

此时，square(y-x)=[[1,4,9,16]，[1,4,9,16]]，各位猜测一下结果是多少呢？没错，就是按照0维坍塌，即“行”，即原来的n行变1行，即按行进行sum，答案为[2,8,18,32]。

举一反三，如果reduction_indices=[1]呢？

 

 

 

 

 

答案是：[30,30],可以理解为[[30],[30]]。

over。