tf.keras.ops.mean

Compute the arithmetic mean along the specified axes.

xInput tensor.
axisAxis or axes along which the means are computed. The default is to compute the mean of the flattened tensor.
keepdimsIf this is set to True, the axes which are reduced are left in the result as dimensions with size one.

Output tensor containing the mean values.