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?
Thank you! We'll create more content like this.
Thank you for helping us improve!