torch.eye
Posted btschang
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了torch.eye相关的知识，希望对你有一定的参考价值。
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]
以上是关于torch.eye的主要内容，如果未能解决你的问题，请参考以下文章