Function moments

  • moments(x, axis?, keepDims?): {
        mean: Tensor;
        variance: Tensor;
    }

  • Calculates the mean and variance of x. The mean and variance are calculated by aggregating the contents of x across axes. If x is 1-D and axes = [0] this is just the mean and variance of a vector.

    Parameters

    • x: Tensor<Rank> | TensorLike

      The input tensor.

    • Optional axis: number | number[]

      The dimension(s) along with to compute mean and variance. By default it reduces all dimensions.

    • Optional keepDims: boolean

      If true, the moments have the same dimensionality as the input.

    Returns {
        mean: Tensor;
        variance: Tensor;
    }

    An object with two keys: mean and variance.

    Doc

Settings

Member Visibility

Theme

Generated using TypeDoc