• Computes the mean of elements across dimensions of a tf.Tensor.

    Reduces x along the dimensions given in axis. Unless keepDims is true, the rank of the tf.Tensor is reduced by 1 for each entry in axis. If keepDims is true, the reduced dimensions are retained with length 1. If axis has no entries, all dimensions are reduced, and a tf.Tensor with a single element is returned.

    const x = tf.tensor1d([1, 2, 3]);

    x.mean().print(); // or tf.mean(a)
    const x = tf.tensor2d([1, 2, 3, 4], [2, 2]);

    const axis = 1;
    x.mean(axis).print(); // or tf.mean(x, axis)

    Type Parameters

    Parameters

    • x: Tensor<Rank> | TensorLike

      The input tensor.

    • Optional axis: number | number[]

      The dimension(s) to reduce. By default it reduces all dimensions.

    • Optional keepDims: boolean

      If true, retains reduced dimensions with size 1.

    Returns T

    Doc

Generated using TypeDoc