摘要: x.dim(dim=0, keepdim=True) 0是按列,1是按行 keepdim是保留行和列这两个维度 阅读全文
posted @ 2020-06-10 10:31 6+0 阅读(127) 评论(0) 推荐(0) 编辑