now in 2024, pytorch has been changed a lot. this 5 years old implementation shows many `deprecation` warnings and..

[shongscience]

some operations got freezed.

Could you check your code for recent pytorch 4.x.x ?!?