Awesome!!! Thanks a lot @Youyoun !! The question has been completely resolved. Why didn’t I think that the mechanism would be different between the Pytorch and Numpy…? Thank you so much. I got a lot of inspiration.