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

arg_nonzero_shape

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

Computes the output shape for the mo.arg_nonzero graph op.

Args:

input_buffer (ManagedTensorSlice[IOSpec[_, _].Input, static_spec=input_buffer.static_spec]): Input tensor whose nonzero element indices are returned.

Returns:

IndexList[Int(2)]: The output shape as a rank-2 IndexList of nonzero element indices.