Good post. I think you mixed torch.eye with torch.full though

You're right! It's wrongly labelled on the image. Thank you for letting me know. Will fix it.