/TensorFlow 1.15


View source on GitHub

Mean of a tensor, alongside the specified axis.

x A tensor or variable.
axis A list of integer. Axes to compute the mean.
keepdims A boolean, whether to keep the dimensions or not. If keepdims is False, the rank of the tensor is reduced by 1 for each entry in axis. If keepdims is True, the reduced dimensions are retained with length 1.
A tensor with the mean of elements of x.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.