nvidia.dali.fn.reductions.variance#

nvidia.dali.fn.reductions.variance(__data, __mean, /, *, axes=None, axis_names=None, bytes_per_sample_hint=[0], ddof=0, keep_dims=False, preserve=False, device=None, name=None)#

获取沿指定轴的元素的方差。

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

  • __data (TensorList) – 运算符的输入。

  • __mean (float 或 TensorList of float) – 用于计算的平均值。

关键字参数:

  • axes (int 或 list of int, 可选) –

    执行归约的轴或多个轴。

    接受范围为 [-ndim, ndim-1]。负索引从末尾计数。

    不提供任何轴将导致所有元素的归约。

  • axis_names (layout str, 可选) –

    执行归约的轴或多个轴的名称。

    输入布局用于将轴名称转换为轴索引，例如，带有输入布局 “FHWC” 的 axis_names="HW" 等效于指定 axes=[1,2]。此参数不能与 axes 一起使用。

  • bytes_per_sample_hint (int 或 list of int, 可选, 默认 = [0]) –

    输出大小提示，以每样本字节数为单位。

    如果指定，则将预先分配位于 GPU 或页锁定主机内存中的运算符的输出，以容纳此大小的样本批次。

  • ddof (int, 可选, 默认 = 0) – 自由度增量。调整计算中使用的除数，即 N - ddof。

  • keep_dims (bool, 可选, 默认 = False) – 如果为 True，则保持原始输入维度。

  • preserve (bool, 可选, 默认 = False) – 即使不使用运算符的输出，也阻止从图中删除该运算符。

在本页中