我有 2 个张量.size of torch.Size([2272, 161])。我想得到它们之间的均方误差。然而，我希望它沿着 161 个通道中的每一个通道，这样我的误差张量就有.size of torch.Size([161])。我怎样才能做到这一点？
看起来torch.nn.MSELoss不允许我指定尺寸。
For the nn.MSELoss你可以指定选项reduction='none'。然后，这将返回两个张量的每个入口位置的平方误差。然后你可以应用torch.sum/torch.mean。
a = torch.randn(2272,161)
b = torch.randn(2272,161)
loss = nn.MSELoss(reduction='none')
loss_result = torch.sum(loss(a,b),dim=0)
我认为没有直接的方法来指定在损失初始化时将平均值/总和应用于哪个维度。希望有帮助！
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
