Combinators
Functions
reduce
Link copied to clipboard
fun reduce(x: FloatTensor, f: (Float, Float) -> Float, axes: IntArray = x.allAxes, keepDims: Boolean = false): FloatTensor
Content copied to clipboard
Reduces x along axes by applying function f to reduce the elements along the axes. If keepDims is true, the original rank of x is retained, otherwise the reduced axes are removed in the output tensor.