Contents

product(alongAxes:keepRank:)

Returns the product along the specified axes.

Declaration

func product(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.product(alongAxes: 0)
y.shape // is [1, 2]
await y.shapedArray(of: Float.self) // is [48.0 105.0]

See Also

Accessing the product along an axes