10-pytorch中对于tensor张量切换维度a.permute()方法

最新推荐文章于 2023-08-24 21:24:32 发布

大鱼不吃鱼（wanna）

最新推荐文章于 2023-08-24 21:24:32 发布

阅读量374

点赞数

分类专栏： Python和机器学习文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/qq_41286751/article/details/126943494

版权

Python和机器学习专栏收录该内容

24 篇文章 2 订阅

订阅专栏

举例： [256, 16, 3,224, 224] -->[256, 3,16,224, 224]

import torch

video_clips=torch.empty(256, 16, 3,224, 224)
print("输出前video , label的size：", video_clips.size())

# 使用permute函数，重新排布位置

video_clips = video_clips.permute(0,2,1,3,4)
print("输出video , label的size：", video_clips.size())