2021年11月21日
Pytorch CrossEntropyLoss 用例
摘要： 交叉熵计算损失 import torch loss_func = torch.nn.CrossEntropyLoss() v1 = torch.tensor([[0.1, 0.7, 0.2]]) v2 = torch.tensor([[0.2, 0.3, 0.5]]) v3 = torch.tens 阅读全文
posted @ 2021-11-21 14:58 太晓 阅读(51) 评论(0) 推荐(0) 编辑