mean(axes:keepDimensions:)
Adds a mean reduction operation along the given axis operation to the current graph.
Declaration
func mean(axes: [Int], keepDimensions: Bool) -> BNNSGraph.Builder.Tensor<T>Parameters
- axes:
The axes over which the operation computes the mean.
- keepDimensions:
A Boolean value that specifies that the operation keeps the reduced dimension with a size of
1.