Iter #10: tensor([[0.8344, 0.1656]], grad_fn=<SoftmaxBackward>)
