@ptrblck sir,
The first line splits the x array (n,64) along the n axis based on indices given in rowSplits and converts it into a raggedTensor of size (32768,[None],64) where None in shape indicates variable number
eg. raggedTensor [0].shape=[20,64] ,raggedTensor [1].shape=[19,64] etc.
The 2nd line takes the Max along axis 1( the variable length axis) to get the output. In the example given here, op[0] is the element wise maximum across the 20 samples obtained. op[1] is for 19 samples etc.
I hope I have made it clearer.
Can I replicate this behaviour using scatter_max or any other operation?
@albanD can you please suggest something on this?
Thank you!