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