albanD:

Hi,
I guess that when you run torch.arange(start, end) , end is smaller than start and so since the default step between them is 1, this is invalid. You should either make the step -1 or make sure that start < end .


Hi, thanks for answering me so fast,
So what would you do specifically? What should I change?