which.max for each row
row_which_max(Y)
Y
a numeric matrix
a integer vector indicating the location of max value in each row