1. 学习
    2. /
  2. 课程
    3. /
  3. PyTorchで学ぶIntroduction to Deep Learning
Connected

练习

MSELoss を使う

回帰問題では、損失関数として交差エントロピーの代わりに Mean Squared Error (MSE) をよく使います。MSEは、予測値（y_pred）と実際の値（y）の差の二乗を計算します。ここでは、NumPy と PyTorch の両方で MSE 損失を計算します。

torch、numpy（np）、および torch.nn（nn）パッケージはすでにインポートされています。

说明

100 XP
  • NumPy を使って MSE 損失を計算します。
  • PyTorch を使って MSE 損失関数を作成します。
  • y_pred と y をテンソルに変換し、mse_pytorch として MSE 損失を計算します。