Hello @Jami_Anwesh_Kumar it looks like that your input ‘x’ is on cpu while the rest of the tensors are on gpu, which is causing this error. Can you please cross check this
Plus you can paste all your code within three quotes