pytorch中维度扩展和维度交换-expand,repeat,permute,transpose
维度扩展expanda=torch.randn(1,32,1,1)a.expand(4,32,14,14)a.shapetorch.Size([4,32,14,14])expand操作是在将该维度扩展成相应的维度,基于原来数据的基础上,推荐使用!注意,要扩展的维度必须是1维的!repeata=torch.randn(1,32,1,1)a.repeat(4,32,14,14).shapea.shap