获取概率张量和索引张量之间的交叉熵损失

Question

计算形状为 x 的概率张量 (N,C) 和形状为 y 的索引张量 N 之间的熵损失的最佳方法是什么？ y 包含正确的类索引（例如，y[0] = 3 表示第 4 个类是 x[0] 的正确类），希望在范围 (N) 中为 i 获得 -log x[i,y] 的总和。< /p>

考虑过列表理解，例如[-torch.log(i[j]) for i,j in zip(x,y)]，但是在使用 torch.cat() 将此列表转换为张量时，它会出错并且会话崩溃。