sum(dim)的使用

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 gt 32 50176 64 1 32 50176 64 3 sum 1

大家好,我是讯享网,很高兴认识大家。
 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)尺寸的数组


讯享网

小讯
上一篇 2025-01-08 15:58
下一篇 2025-01-11 16:42

相关推荐

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容,请联系我们,一经查实,本站将立刻删除。
如需转载请保留出处:https://51itzy.com/kjqy/126266.html