PyTorch torch.var_mean 函数
Pytorch torch 参考手册torch.var_mean 是 PyTorch 中用于同时返回张量的方差和均值的函数。这是一个高效的函数，可以在一次操作中同时计算方差和均值。
函数定义
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)
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])