Thankyou so much for the reply!, i was also wondering how do i pass these values to nn.CrossEntropy function. Won’t i get an dimensional mismatch?
And also wanted to know after getting the mask from your code, can i directly pass these masks from the get_item() function?