PyTorch reduction的作用是什么

这篇文章主要讲解了"PyTorch reduction的作用是什么"，文中的讲解内容简单清晰，易于学习与理解，下面请大家跟着小编的思路慢慢深入，一起来研究和学习"PyTorch reduction的作用是什么"吧！

损失函数的reduction有三种模式，它们的作用分别是什么？

当inputs和target及weight分别如以下参数时，reduction='mean'模式时，loss是如何计算得到的？

inputs = torch.tensor([[1, 2], [1, 3], [1, 3]], dtype=torch.float)

target = torch.tensor([0, 1, 1], dtype=torch.long)

weights = torch.tensor([1, 2]

加权交叉熵 Loss

import torchimport torch.nn as nninputs = torch.tensor([[1, 2], [1, 3], [1, 3]], dtype=torch.float)target = torch.tensor([0, 1, 1], dtype=torch.long)# def loss functionweights = torch.tensor([1, 200], dtype=torch.float)loss_f_none_w = nn.CrossEntropyLoss(weight=weights, reduction='none')loss_f_sum = nn.CrossEntropyLoss(weight=weights, reduction='sum')loss_f_mean = nn.CrossEntropyLoss(weight=weights, reduction='mean')# forwardloss_none_w = loss_f_none_w(inputs, target)loss_sum = loss_f_sum(inputs, target)loss_mean = loss_f_mean(inputs, target)# viewprint("\nweights: ", weights)print(loss_none_w, loss_sum, loss_mean)

感谢各位的阅读，以上就是"PyTorch reduction的作用是什么"的内容了，经过本文的学习后，相信大家对PyTorch reduction的作用是什么这一问题有了更深刻的体会，具体使用情况还需要大家实践验证。这里是，小编将为大家推送更多相关知识点的文章，欢迎关注！