input = torch.randn(20, 2, 11, 11, 32)输出形式

黯止依蓝2023-12-15 19:05

复制代码

input = torch.randn(20, 2, 11, 11, 32)
m = torch.nn.AdaptiveAvgPool3d((1,1, 32))
x=m(input)
print(x.shape)

结果：

也就是不用管批次和通道数

上一篇：创建链表时的一个小bug

下一篇：【NR技术】NR NG-RAN整体架构 -网络接口以及无线协议框架（四）

热门推荐

01国内可直接用、免费额度/永久免费的大模型API清单（含 SiliconFlow、火山、阿里、智谱、百度、Kimi、DeepSeek、DMXAPI 等）02如何新建文件夹？电脑新建文件夹的4种方法 03历年考研数学一、数学二、数学三真题试卷及答案PDF 04几个好用的ip纯净度检测网站 05微信历史版本含下载地址（ Windows PC | 安卓 | MAC ）及设置微信不更新 06昇腾910B系列芯片完全指南：四款型号对比、服务器选型与大模型部署实战 07GitHub 镜像站点 08苹果iPA游戏软件资源下载网站 09蜘蛛磁力搜索引擎大全，如何使用蜘蛛磁力查找磁力链接 10Dify：一站式AI应用开发平台全解析与Windows部署指南