Mojo function
argmin
argmin(input: TileTensor[dtype, LayoutType, origin, address_space=address_space, linear_idx_type=linear_idx_type, element_shape_types=element_shape_types], axis: Int, output: TileTensor[dtype, LayoutType, origin, address_space=address_space, linear_idx_type=linear_idx_type, element_shape_types=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[dtype, LayoutType, origin, address_space=address_space, linear_idx_type=linear_idx_type, element_shape_types=element_shape_types], axis_buf: TileTensor[dtype, LayoutType, origin, address_space=address_space, linear_idx_type=linear_idx_type, element_shape_types=element_shape_types], output: TileTensor[dtype, LayoutType, origin, address_space=address_space, linear_idx_type=linear_idx_type, element_shape_types=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!