Applying Smote technique (class imbalance) in Pytorch?

Hello,

I want to apply Smote technique on my training data . I got following error. Can anyone asses me to solve this error.

Thank you