Contents

max(alongAxes:keepRank:)

Returns the maximum values along the specified axes.

Declaration

func max(alongAxes axes: Int..., keepRank: Bool = false) -> MLTensor

Parameters

  • axes:

    The axes to reduce.

  • keepRank:

    A Boolean indicating whether to keep the reduced axes or not. The default value is false.

Return Value

The reduced tensor.

Discussion

let x = MLTensor(shape: [3, 2], scalars: [2, 3, 4, 5, 6, 7], scalarType: Float.self)
let y = x.max(alongAxes: 0, keepRank: true)
y.shape // is [1, 2]
await y.shapedArray(of: Float.self) // is [6.0 7.0]

See Also

Accessing the minimum, maximum and mean