获取两个张量中每个元素之间的均方误差
Get mean squared error between every element in two tensors
我有两个单维张量，y_pred 和 y_true
其中：
>>> y_pred.shape
torch.Size([2730441, 1])
>>> y_true.shape
torch.Size([2730441, 1])
要获得我可以使用的两个张量之间的均方误差
torch.nn.MSELoss() 但是，我想得到张量 y_pred 和 y_true 中每一行/元素之间的损失，即我想 运行 一些函数 elementWiseMSE(y_pred, y_true) 这将return loss_tensor 形状 [2730441, 1] 其中包含所有预测的元素平均误差。
你要找的“功能”就是字面意思
loss_tensor = (y_pred - y_true) ** 2
我有两个单维张量，y_pred 和 y_true
其中：
>>> y_pred.shape
torch.Size([2730441, 1])
>>> y_true.shape
torch.Size([2730441, 1])
要获得我可以使用的两个张量之间的均方误差
torch.nn.MSELoss() 但是，我想得到张量 y_pred 和 y_true 中每一行/元素之间的损失，即我想 运行 一些函数 elementWiseMSE(y_pred, y_true) 这将return loss_tensor 形状 [2730441, 1] 其中包含所有预测的元素平均误差。
你要找的“功能”就是字面意思
loss_tensor = (y_pred - y_true) ** 2