Mojo function

argmin

argmin(input: TileTensor[input.dtype, input.LayoutType, input.origin, address_space=input.address_space, linear_idx_type=input.linear_idx_type, element_shape_types=input.element_shape_types], axis: Int, output: TileTensor[output.dtype, output.LayoutType, output.origin, address_space=output.address_space, linear_idx_type=output.linear_idx_type, element_shape_types=output.element_shape_types])

Finds the indices of the minimum element along the specified axis.

Args:

• ​input (TileTensor): The input tensor.
• ​axis (Int): The axis.
• ​output (TileTensor): The output tensor.

argmin(input: TileTensor[input.dtype, input.LayoutType, input.origin, address_space=input.address_space, linear_idx_type=input.linear_idx_type, element_shape_types=input.element_shape_types], axis_buf: TileTensor[axis_buf.dtype, axis_buf.LayoutType, axis_buf.origin, address_space=axis_buf.address_space, linear_idx_type=axis_buf.linear_idx_type, element_shape_types=axis_buf.element_shape_types], output: TileTensor[output.dtype, output.LayoutType, output.origin, address_space=output.address_space, linear_idx_type=output.linear_idx_type, element_shape_types=output.element_shape_types])

Finds the indices of the minimum element along the specified axis.

Args:

• ​input (TileTensor): The input tensor.
• ​axis_buf (TileTensor): The axis tensor.
• ​output (TileTensor): The axis tensor.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit