我正在尝试建立一个损失函数,它将计算相同大小的两个男高音的均方误差。换句话说,我需要一个函数来计算矩阵A和矩阵B上每2个单元格(具有相同的行和列)的差值,将其平方并计算差值的平均值。据我所知,nn.MSELoss应该做到这一点。当我将2个张量传递给nn.MSELoss时,我得到了以下错误消息: RuntimeError: Boolean value of Tensor with more than
.size of torch.Size([2272, 161])有两个张量。我想得到他们之间的均方误差。但是,我希望沿着161个通道中的每个通道,这样我的错误张量就有一个.size of torch.Size([161])。我怎样才能做到这一点?
torch.nn.MSELoss似乎不允许我指定维度。