For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Mojo function

reduce_mul_shape

def reduce_mul_shape[input_rank: Int, input_type: DType, axis: Int](input: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=input.static_spec]) -> IndexList[input_rank]

Computes the output shape for the mo.reduce.mul graph op.

Parameters:

input_rank (Int): Number of dimensions in the input and output tensors.
input_type (DType): Element type of the input tensor.
axis (Int): Dimension along which to multiply the elements of input.

Args:

input (ManagedTensorSlice[IOSpec[_, _].Input, static_spec=input.static_spec]): Input tensor reduced along axis by multiplication.

Returns:

IndexList[input_rank]: The output shape after reducing input along axis by multiplication.