小白学Pytorch系列--Torch.nn API (2)
创始人
2025-05-31 22:30:02

小白学Pytorch系列–Torch.nn API (2)

Convolution Layers

方法注释
nn.Conv1d对由多个输入平面组成的输入信号应用1D卷积。
nn.Conv2d对由多个输入平面组成的输入信号应用2D卷积。
nn.Conv3d对由多个输入平面组成的输入信号应用3D卷积。
nn.ConvTranspose1d对由多个输入平面组成的输入图像应用一维转置卷积算子。
nn.ConvTranspose2d在由多个输入平面组成的输入图像上应用二维转置卷积算子。
nn.ConvTranspose3d在由多个输入平面组成的输入图像上应用三维转置卷积算子。
nn.LazyConvTranspose1dtorch.nn.Conv1d模块,对Conv1din_channels参数进行惰性初始化,该参数是从input.size(1)推断出来的。
nn.LazyConvTranspose2dtorch.nn.Conv2d模块,从input.size(1)推断Conv2din_channels参数进行延迟初始化。
nn.LazyConvTranspose3dtorch.nn.Conv3d模块,从input.size(1)推断Conv3din_channels参数进行延迟初始化。
nn.Unfold从批处理输入张量中提取滑动局部块。
nn.Fold将一个滑动局部块数组合并为一个大的包含张量。

nn.Conv1d

nn.Conv2d

nn.Conv3d

nn.ConvTranspose1d

nn.ConvTranspose2d

nn.ConvTranspose3d

nn.LazyConvTranspose1d

nn.LazyConvTranspose2d

nn.LazyConvTranspose3d

nn.Unfold

nn.Fold

相关内容

热门资讯

今年我省粮食产量达515.56... (来源:辽宁日报)转自:辽宁日报 图为在中储粮(盘锦)储运有限公司,装运粮食的重型卡车排起长队...
国家发展改革委部署促进投资止跌... (来源:辽宁日报)转自:辽宁日报 新华社北京12月13日电 (记者魏玉坤) 记者13日从全国发展和改...
江苏省实施《中华人民共和国森林... (来源:新华日报) 目 录 第一章 总则 第二章 森林、林木和林地权属管理...
姜堰数字化产品讲“活”理论 (来源:新华日报) □ 本报记者 卢佳乐 通讯员 姜宣 “王教授,您约我‘喝茶论道’,...
联合国维和部队在苏丹遇袭 6人... 转自:财联社【联合国维和部队在苏丹遇袭 6人死亡】财联社12月14日电,当地时间13日,苏丹武装部队...