For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo function
arg_nonzero_shape
def arg_nonzero_shape(input_buffer: T) -> IndexList[Int(2)]
Computes the output shape for the mo.arg_nonzero graph op.
Args:
- input_buffer (
T): Input tensor whose nonzero element indices are returned.
Returns:
IndexList[Int(2)]: The output shape as a rank-2 IndexList of nonzero element indices.