chenyuntc:

C=A[:,B,1]


I think I didn’t get what you meant by this (it’s not working piece of code thus I needed to guess). What is the one line of code required to be preceded and what does it supposed to do with this situation?
At first, I thought you were saying that I should allocate new memory for C (that is with shape: 64 10 256) before putting computed output but I think it wasn’t a point as the error remains intact. Is it because I overly used bunch of .repeat() operations in my model?
Or should I put torch.cuda.synchronize()? after each suspected operations?
Trying not to be too annoying but I think I met a giant wall in front of me please help  );