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