Iter #10: tensor([[0.0263, 0.9737]], device='cuda:0', grad_fn=<SoftmaxBackward>)
Iter #20: tensor([[0.0060, 0.9940]], device='cuda:0', grad_fn=<SoftmaxBackward>)
