Skip to main content
Log in

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?