Little confusion in reading the code

[Andrewsama]

Hello, Thanks for your work which help me receive some inspiration. I'm confused when I read the code. I can't understand the code " batch = torch.LongTensor( np.repeat( np.arange( batch.shape[0]), max_nodes)).cuda()" on 'mvgrl/train.py' in line 272. I am more inclined to the code like " batch = torch.LongTensor( np.repeat( batch, max_nodes)).cuda()". Maybe I'm not thoughtful, I truly want your help, thx.