如何对张量的列求和？
torch.Size([10, 100]) ---> torch.Size([10])
最简单和最好的解决方案是使用torch.sum().
对张量的所有元素求和：
torch.sum(x) # gives back a scalar
对所有行（即每列）求和：
torch.sum(x, dim=0) # size = [ncol]
对所有列（即每一行）求和：
torch.sum(x, dim=1) # size = [nrow]
应该注意的是，相加的维度将从结果张量中消除。
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
