PyTorch | 通过torch.eye创建单位对角矩阵 | torch.eye()如何使用？torch.eye()例子 | torch.eye()使用方法

在公众号【计算机视觉联盟】后台回复【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使用方法

PyTorch | torch.randperm()使用方法

PyTorch | torch.tensor使用方法

PyTorch | torch.zeros()使用方法

PyTorch | torch.full()使用方法

PyTorch | torch.arange()使用方法

PyTorch | torch.linspace()使用方法