小白学习之pytorch框架(4)-softmax回归(torch.gather()、torch.argmax())