2020年3月10日
Pytorch之permute函数
摘要： 1、主要作用：变换tensor维度 example： import torch x = torch.randn(2, 3, 5) print(x.size()) print(x.permute(2, 0, 1).size()) >>>torch.Size([2, 3, 5]) >>>torch.Si 阅读全文
posted @ 2020-03-10 23:01 嶙羽 阅读(4951) 评论(0) 推荐(0) 编辑