Why is the shape of the weight different?(Pytorch)

Question

Im trying to to do SoftMaxRegression in 2ways

First Way:

x_train = torch.FloatTensor([[1, 2, 1, 1],
           [2, 1, 3, 2],
           [3, 1, 3, 4],