Hi, is there any Pytorch function to check which operation produces nan?
Yes! Have a look at the Anomaly Detection docs.