I have input tensor (N*2) , first column is category and second is the value, something like this:
[ [14,50] , [18,1.5] , [14,250] , [18,2.5] , [10,1] , [14,252] , [18,5.3] ]
each category may be have N row in input tensor,
how to found index of row in input tensor that have max value in each category, something like this for above input tensor
[ [14,5] , [18,6] , [10,4] ]
[ category code, row index have max value ]
but if it’s possible without using loop,
thank’s