问题来源:

a = torch.zeros(2, 3)
for i in range(2):
    for j in range