- 博客(1)
- 收藏
- 关注
原创 实现 pytorch 中 torch.nn.CrossEntropyLoss
简介为了更好地理解 pytorch 的 CrossEntropyLoss,于是打算进行简单的实现。官方文档:https://pytorch.org/docs/stable/nn.html?highlight=crossentropyloss#torch.nn.CrossEntropyLoss官网 loss 的公式:x 的维度是 (batch_size, C)class 的维度是 (b...
2019-08-13 16:53:06 5135
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人