torch.std

torch.std(input, dim=None, *, correction=1, keepdim=False, out=None) → Tensor

计算由 dim 指定的维度上的标准差。 dim 可以是单个维度、维度列表或 None，以对所有维度进行缩减。

标准差 ($\sigma$) 计算如下

$$\sigma = \sqrt{\frac{1}{\max(0,~N - \delta N)}\sum_{i=0}^{N-1}(x_i-\bar{x})^2}$$

其中 $x$ 是样本元素集，$\bar{x}$ 是样本均值，$N$ 是样本数量，$\delta N$ 是 correction。

如果 keepdim 为 True，则输出张量与 input 的大小相同，除了 dim 维度，该维度的大小为 1。否则，dim 被压缩（参见 torch.squeeze()），导致输出张量少了一个（或 len(dim) 个）维度。

参数

• input (张量) – 输入张量。

• dim (整数 或 元组 的 整数) – 要缩减的维度或维度。

关键字参数

• correction (整数) –

  样本大小和样本自由度之间的差。默认为 贝塞尔校正，correction=1。

  在版本 2.0 中更改: 之前此参数称为 unbiased，并且是一个布尔值，其中 True 对应于 correction=1，而 False 为 correction=0。

• keepdim (布尔值) – 输出张量是否保留 dim。

• out (张量, 可选) – 输出张量。

示例

>>> 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.std(a, dim=1, keepdim=True)
tensor([[1.0311],
        [0.7477],
        [1.2204],
        [0.9087]])