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]])