需要鼓励

摘要： 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 例子:... 阅读全文