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_row_mean_of_squares_shape

def reduce_row_mean_of_squares_shape(input: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=input.static_spec]) -> IndexList[Int(2)]

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

Args:

input (ManagedTensorSlice[IOSpec[_, _].Input, static_spec=input.static_spec]): Two-dimensional input tensor of shape [M, N] whose per-row mean of squares produces a [M, 1] output.

Returns:

IndexList[Int(2)]: The output shape [M, 1] after computing the per-row mean of squares.