pytocrh学习记录0507。

一。toch.nn.CELoss
最近在使用该loss的时候遇到了如下三个问题。

1. One-hot问题。
   RuntimeError: multi-target not supported at /opt/conda/conda-bld/pytorch_1587428398394/work/aten/src/THCUNN/generic/ClassNLLCriterion.cu:18
   参考此篇博客.这是因为CEloss期望的真值输入不能是one-hot形式的，打印一下原本的预测值和真值维度、部分数据如下：

   pred shape: torch.Size([128, 10])
   target shape: torch.Size([128, 10])
   pred: tensor([ -1.8647, 11.9629, -8.9586, -4.3062, -2.7667, 2.8557, -1.5226,-12.8641, 3.2789, 0.8572], device=‘cuda:0’, grad_fn=)
   target: tensor([0., 0., 0., 0., 1., 0., 0., 0., 0., 0.], device=‘cuda:0’)

CELoss在内部运算过程中会自动进行one-hot编码，所以，这里target中对应维度应该是【128， 1】， 具体到上述打印中的例子， 应该就是单维张量tensor(4, device='cuda:0')。
2. 数据类型问题

RuntimeError: Expected object of scalar type Long but got sca