Locate the position of the maximum value

rray_max_pos() returns the integer position of the maximum value over an axis.

rray_max_pos(x, axis = NULL)

Arguments

x	A vector, matrix, array, or rray.
axis	A single integer specifying the axis to compute along. 1 computes along rows, reducing the number of rows to 1. 2 does the same, but along columns, and so on for higher dimensions. The default of NULL first flattens x to 1-D.

Value

An integer object of the same type and shape as x, except along axis, which has been reduced to size 1.

Examples

x <- rray(c(1:10, 20:11), dim = c(5, 2, 2))

# Find the max position over all of x
rray_max_pos(x)
#> <rray<int>[,1,1][1]>
#> , , 1
#> 
#>      [,1]
#> [1,]   11
#> 

# Compute along the rows
rray_max_pos(x, 1)
#> <rray<int>[,2,2][1]>
#> , , 1
#> 
#>      [,1] [,2]
#> [1,]    5    5
#> 
#> , , 2
#> 
#>      [,1] [,2]
#> [1,]    1    1
#> 

# Compute along the columns
rray_max_pos(x, 2)
#> <rray<int>[,1,2][5]>
#> , , 1
#> 
#>      [,1]
#> [1,]    2
#> [2,]    2
#> [3,]    2
#> [4,]    2
#> [5,]    2
#> 
#> , , 2
#> 
#>      [,1]
#> [1,]    1
#> [2,]    1
#> [3,]    1
#> [4,]    1
#> [5,]    1
#>