tensor([-2.0770, -2.2102, -2.5764, -1.5185, -2.5472, -2.3393, -2.4241, -2.8882,
        -2.7545, -2.4377], grad_fn=<SelectBackward0>) torch.Size([64, 10])