04 2022 档案

浅谈矩阵操作中axis=i的理解
摘要：不妨设一个矩阵(Matrix) A , 那么对A进行axis=i,即将张量A的第i个维度拍扁后得到的张量 如: A = torch.tensor([[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3]]) A.sum(axis = 0) 得到结果: tensor([6, 阅读全文
posted @ 2022-04-27 21:10 Krasitz 阅读(195) 评论(0) 推荐(0)