mse_loss関数【torch】

説明：平均二乗誤差を求める関数。

書式：

torch.nn.functional.mse_loss( 引数１、引数２)

引数：データを引数１，２の間の平均二乗誤差をテンソルで与える。

戻り値：平均二乗誤差を表すテンソル

使い方：

code:mselos1s.py

import torch.nn.functional as F

import torch as pt

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

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

loss = F.mse_loss(data, target)

print(loss)

備考：

import torch.nn.functional as F

のFは慣習的に用いられる略記であり、特に指定があるわけではない。

code:mselos1s.py

import torch

import torch as pt

import torch.nn.functional as hoge

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

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

loss_func = torch.nn.functional.mse_loss(data, target)

loss1 = hoge.mse_loss(data, target)

loss2 = torch.nn.functional.mse_loss(data, target)

print(loss1)

print(loss2)

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