Computes the mean of the values of a Tensor over the whole dataset.

Used in the notebooks

Used in the tutorials

x A Tensor, SparseTensor, or RaggedTensor. Its type must be floating point (float{16|32|64}), or integral ([u]int{8|16|32|64}).
reduce_instance_dims By default collapses the batch and instance dimensions to arrive at a single scalar output. If False, only collapses the batch dimension and outputs a vector of the same shape as the input.
name (Optional) A name for this operation.
output_dtype (Optional) If not None, casts the output tensor to this type.

A Tensor containing the mean. If x is floating point, the mean will have the same type as x. If x is integral, the output is cast to float32. NaNs and infinite input values are ignored.

TypeError If the type of x is not supported.