mlx.core.var

目录

mlx.core.var#

var(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, ddof: int = 0, *, stream: None | Stream | Device = None) array#

计算给定轴上的方差。

参数:
  • a (array) – 输入数组。

  • axis (intlist(int), 可选) – 要在其上执行约简的轴或轴的列表。如果未指定,则默认为对整个数组进行约简。

  • keepdims (bool, 可选) – 是否保留约简后的轴作为单维度,默认为 False

  • ddof (int, 可选) – 计算方差时使用的除数是 N - ddof,默认为 0。

返回:

方差的输出数组。

返回类型:

array