Dear Ptrblck,
I am so sorry for I did not explain well.
I want to try this function through Pytorch.

So in a matrix, I want to classify numbers of different size ranges.
In this case, if I still don’t explain well, could you let me know? I will try again.
Thank you for your time.