2019年1月21日
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 例子:... 阅读全文
posted @ 2019-01-21 20:13 可爱小小畅 阅读(5544) 评论(0) 推荐(0) 编辑