megengine.functional.std

std(inp, axis=None, keepdims=False)[源代码]

返回在给定轴 axisinp 张量每一行的标准差。如果给定的轴是一个维度列表,则规约所有维度。

参数
  • inp (Tensor) – 输入张量。

  • axis (Union[int, Sequence[int], None]) – 要进行规约的维度。如果设置为 None,则所有的维度都将规约。默认:None

  • keepdims (bool) – 输出张量是否保留了轴 axis 。默认:False

返回类型

Tensor

返回

输出张量。

实际案例

>>> import numpy as np
>>> data = Tensor(np.arange(1, 7, dtype=np.float32).reshape(2, 3))
>>> out = F.std(data, axis=1)
>>> out.numpy().round(decimals=4)
array([0.8165, 0.8165], dtype=float32)