Comment Comment by lappa Good post. I think you mixed torch.eye with torch.full though parent lappa 8 months ago