Skip to main content

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.

Was this page helpful?