torch.eye

torch.eye

torch.eye(n, m=None, out=None)

返回一個2維張量，對角線位置全1，其它位置全0

• n (int ) – 行數
• m (int, optional) – 列數.如果為None,則默認為n
• out (Tensor, optinal) - Output tensor

返回值: 對角線位置全1，其它位置全0的2維張量

返回值類型: Tensor

例子:

>>> torch.eye(3)

1 0 0

0 1 0

0 0 1

[torch.FloatTensor of size 3x3]