IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /max/get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
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 module

arg_nonzero

Functionsโ€‹

  • โ€‹arg_nonzero: Gather the indices of all non-zero elements in input buffer storing the indices in the output_buffer.
  • โ€‹arg_nonzero_shape: Return [NumNonZeros, InputRank] where NumNonZeros are the number of non-zero elements in the input.