- 博客(2)
- 收藏
- 关注
原创 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
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人