Iter #10: tensor([[0.9627, 0.0373]], grad_fn=<SoftmaxBackward>)
