torch.eye()
宋安晏
x = torch.eye(3)
输出：一个3 x 3的矩阵 对角线全为1.
参数：torch.eye(行，列，tensor）
类似资料：
x = torch.eye(3)
输出：一个3 x 3的矩阵 对角线全为1.
参数：torch.eye(行，列，tensor）