摘要: permute的中文含义是:置换 这个就已经解释了一半了。 他的功能就是把一个多维张量的维度进行调换。 对于一个2维张量来说,例如: a=torch.rand(2,3). 那么代码a.permute(1,0)就相当于a.T 对于一个三维张量来说,例如: a=torch.rand(2,3,4) 代码a 阅读全文
posted @ 2024-10-24 11:28 Cooker2008 阅读(517) 评论(0) 推荐(1)