Python 如何计算三维图像的交叉熵?

Python 如何计算三维图像的交叉熵?,python,pytorch,Python,Pytorch,左边是(2480640),它是softmax值 正确的东西是(2480640),它是一个热编码值 如何得到所有元素的交叉熵损失 与任何其他图像的方式完全相同。使用。注意 两者都必须是torch.float32dtype,因此您可能需要首先使用right.to(torch.float32)将right转换为(torch.float32) 如果左张量包含logit而不是概率,那么调用二进制交叉熵(torch.sigmoid(左),右)

左边是(2480640),它是softmax值

正确的东西是(2480640),它是一个热编码值


如何得到所有元素的交叉熵损失

与任何其他图像的方式完全相同。使用。注意

  • 两者都必须是
    torch.float32
    dtype
    ,因此您可能需要首先使用
    right.to(torch.float32)
    right
    转换为(torch.float32)
  • 如果
    张量包含logit而不是概率,那么调用
    二进制交叉熵(torch.sigmoid(左),右)