在公众号【计算机视觉联盟】后台回复【9076】获取独家200页手推AI笔记;我的微信:PursueWin; --by Sophia 中科院学霸 | 上市AI算法工程师 | CSDN博客专家
t = torch.eye(3)
print("torch.eye(3)的输出结果\n",t)
运行结果:
torch.eye(3)的输出结果
tensor([[1., 0., 0.],
[0., 1., 0.],
[0., 0., 1.]])
推荐阅读:
PyTorch | torch.from_numpy使用方法