W

[2698022795]

Hello , I wonder why the code in train() function writing like this?

target = torch.stack([target * 4 + k for k in range(4)], 1).view(-1)

instead of using this

target = torch.stack([target * k for k in range(4)], 1).view(-1)