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)尺寸的数组