MSELoss関数【torch】 - takalab-lecture-ICP

MSELoss関数【torch】

説明：平均二乗誤差による損失関数の値を求めるための関数。nn.Moduleクラスを継承している。

書式：

torch.nn.MSELoss( )

引数：なし

戻り値：MSELoss型インスタンス

備考：MSELoss型のインスタンスは、テンソル型の引数を要求する。

例：

変数dataとtargetの間の誤差を求める。

code:mseloss1.py

import torch.nn as nn

import torch as pt

loss_func = nn.MSELoss() # 損失関数の生成

data1 = pt.tensor(1, 2, 1, dtype=pt.float)

data2 = pt.tensor(1, 2, 0, dtype=pt.float)

loss = loss_func(data1, data2) # 損失関数の計算

print(loss)

関連：クラスと関数【torch】

https://scrapbox.io/files/650a983a194b89001ce5c816.png