Contents

sumOfSquares(axes:keepDimensions:)

Adds a sum-of-squares reduction operation along the given axis operation to the current graph.

Declaration

func sumOfSquares(axes: [Int], keepDimensions: Bool) -> BNNSGraph.Builder.Tensor<T>

Parameters

  • axes:

    The axes over which the operation computes the sum of squares.

  • keepDimensions:

    A Boolean value that specifies that the operation keeps the reduced dimension with a size of 1.