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_size=input.element_size], axis: Int, output: TileTensor[output.dtype, output.LayoutType, output.origin, address_space=output.address_space, linear_idx_type=output.linear_idx_type, element_size=output.element_size])

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_size=input.element_size], 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_size=axis_buf.element_size], output: TileTensor[output.dtype, output.LayoutType, output.origin, address_space=output.address_space, linear_idx_type=output.linear_idx_type, element_size=output.element_size]) where (axis_buf.flat_rank == 1)

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.