torch.eye
torch.eye(n, m=None, out=None)
返回一个2维张量，对角线位置全1，其它位置全0
返回值: 对角线位置全1，其它位置全0的2维张量
返回值类型: Tensor
例子:
>>> torch.eye(3)
1 0 0
0 1 0
0 0 1
[torch.FloatTensor of size 3x3]
原文：https://www.cnblogs.com/btschang/p/10300727.html
