PyTorch torch.var_mean 関数

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.var_meanは、PyTorch においてテンソルの分散と平均を同時に返す関数です。これは、1回の操作で分散と平均を同時に計算できる効率的な関数です。

関数定義

torch.var_mean(input, dim, unbiased, keepdim=False)

使用例

例

import torch

x = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])

# 分散と平均を同時に返す
var, mean = torch.var_mean(x)
print("分散:", var)
print("平均:", mean)

# dim=0 に沿って分散と平均を同時に返す
y = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
var0, mean0 = torch.var_mean(y, dim=0)
print("dim=0 分散:", var0)
print("dim=0 平均:", mean0)

出力結果は次のとおりです：

方差: tensor(2.5000)
均值: tensor(3.)
dim=0 方差: tensor([2.2500, 2.2500, 2.2500])
dim=0 均值: tensor([2.5000, 3.5000, 4.5000])

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張