pytorch框架常用函数总结[自用-持续更新]
torch.permute(dims)调换维度,dims必须match原始tensor的维度>>>x = torch.randn(2,3,4)>>>x.size()Out[5]: torch.Size([2, 3, 4])>>>x.permute(2,0,1).size()Out[6]: torch.Size([4, 2, 3])torch.cat((A,B),axis)对A, B两个tensor进行拼接,参数axis指定拼接的方式。a
原创
2020-07-23 11:34:34 ·
496 阅读 ·
0 评论