1. Learn
    2. /
  2. 课程
    3. /
  3. 使用 PyTorch 的深度学习入门
Connected

道练习

使用 MSELoss

对于回归问题，通常使用均方误差（MSE）作为损失函数，而不是交叉熵。MSE 计算预测值（y_pred）与真实值（y）之间的平方差。现在，您将分别使用 NumPy 和 PyTorch 计算 MSE 损失。

torch、numpy（简称 np）和 torch.nn（简称 nn）包已导入。

说明

100 XP
  • 使用 NumPy 计算 MSE 损失。
  • 使用 PyTorch 创建一个 MSE 损失函数。
  • 将 y_pred 和 y 转换为张量，然后将 MSE 损失计算为 mse_pytorch。