The computational graph is shown as following:
The output of the listing is shown as follows:
New data, epoch 0Step 0 Loss 0.777418Step 600 Loss 0.693907New data, epoch 1Step 0 Loss 0.690996Step 600 Loss 0.691115New data, epoch 2Step 0 Loss 0.69259Step 600 Loss 0.685826New data, epoch 3Step 0 Loss 0.684189Step 600 Loss 0.690608New data, epoch 4Step 0 Loss 0.691302Step 600 Loss 0.691309New data, epoch 5Step 0 Loss 0.69172Step 600 Loss 0.694034New data, epoch 6Step 0 Loss 0.692927Step 600 Loss 0.42796New data, epoch 7Step 0 Loss 0.42423Step 600 Loss 0.00845207New data, epoch 8Step 0 Loss 0.188478Step 500 Loss 0.00427217
