torch.var¶
- torch.var(input, dim=None, *, correction=1, keepdim=False, out=None) Tensor ¶
计算由
dim
指定维度上的方差。dim
可以是单个维度、维度列表,或者None
以在所有维度上进行缩减。方差 () 的计算公式为
其中 是元素的样本集, 是样本均值, 是样本数, 是
correction
。如果
keepdim
为True
,则输出张量的大小与input
相同,只是在维度dim
上的大小为 1。否则,dim
将被压缩(参见torch.squeeze()
),从而使输出张量的维度减少 1 个(或len(dim)
个)。- 参数
- 关键字参数
示例
>>> a = torch.tensor( ... [[ 0.2035, 1.2959, 1.8101, -0.4644], ... [ 1.5027, -0.3270, 0.5905, 0.6538], ... [-1.5745, 1.3330, -0.5596, -0.6548], ... [ 0.1264, -0.5080, 1.6420, 0.1992]]) >>> torch.var(a, dim=1, keepdim=True) tensor([[1.0631], [0.5590], [1.4893], [0.8258]])