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