sum(dim)的使用

  encoded_feat = (assignment_weights * (expanded_x - reshaped_codewords)).sum(1)

#{(32,50176,64,1)[(32,50176,64,3)-(1,1,64,3)].sum(1)}---->{(32,50176,64,1)[(32,50176,64,3).sum(1)}
注意在某维度求和就是对于这个维度的比如x个块当中的同一位置处求和最终生成一个块。而这个块是此维度下属维度的数组成的。比如(32,50176,64,3).sum(1)就是在第二维度进行求和，把（64，3）这个块也就是三四维的数据组成的块，在50176个（64，3）这样的块的相同位置处的数据进行求和输出一个最终的值，最后50176的这个维度也就被压缩没了生成了新的(32,64,3)尺寸的数组