[lappa]

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

[0bytematt]

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