Pytorch data and change the label shape

1： out: torch.Size([128, 5]) torch.Size([128, 1, 5])

2: label.view using () transformation

    print("out:",out.shape,label.shape)
    label = label.view(out.shape[0], out.shape[1])
    print(label.shape)