mean
Mean along axis.
This layer computes mean (average) values along the specified axis.
Parameter
- axes : Integers, axes along which mean is applied.
The default is [-1] (the last dimension). - keepdims : Boolean, whether the tensor rank is saved.
The default is false.