ConvTranspose1d¶
- class torch.nn.ConvTranspose1d(in_channels, out_channels, kernel_size, stride=1, padding=0, output_padding=0, groups=1, bias=True, dilation=1, padding_mode='zeros', device=None, dtype=None)[源代码]¶
对由多个输入平面组成的输入图像应用一维转置卷积运算符。
这个模块可以被视为Conv1d相对于其输入的梯度。 它也被称为分数步长卷积或 反卷积(尽管它不是一个实际的反卷积操作,因为它不计算卷积的真实逆)。更多信息,请参见这里的可视化和反卷积网络论文。
此模块支持 TensorFloat32。
在某些 ROCm 设备上,当使用 float16 输入时,此模块将在反向传播中使用 不同的精度。
stride控制交叉相关的步幅。padding控制两侧隐式零填充的数量,用于dilation * (kernel_size - 1) - padding的点数。详情请参见下文注释。output_padding控制添加到输出形状一侧的额外大小。详情请参见下文。dilation控制卷积核点之间的间距;也称为à trous算法。 这很难描述,但这里有一个很好的可视化展示了dilation的作用。groups控制输入和输出之间的连接。in_channels和out_channels必须都能被groups整除。例如,在 groups=1 时,所有输入都会被卷积到所有输出。
在 groups=2 时,操作变得等同于并排有两个卷积层,每个卷积层看到一半的输入通道并产生一半的输出通道,然后将两者连接起来。
在 groups=
in_channels的情况下,每个输入通道都与自己的一组过滤器(大小为 )进行卷积。
注意
参数
padding实际上在输入的两边添加了dilation * (kernel_size - 1) - padding数量的零填充。这样设置是为了当Conv1d和ConvTranspose1d使用相同的参数初始化时,它们在输入和输出形状方面互为逆。然而,当stride > 1时,Conv1d将多个输入形状映射到相同的输出形状。output_padding通过有效地增加计算出的输出形状的一侧来解决这种歧义。请注意,output_padding仅用于确定输出形状,但实际上并不会在输出上添加零填充。注意
在某些情况下,当使用CUDA后端与CuDNN时,此操作符可能会选择一个非确定性算法以提高性能。如果这是不可取的,您可以尝试通过设置
torch.backends.cudnn.deterministic = True来使操作具有确定性(可能会以性能为代价)。 请参阅可重复性的背景说明。- Parameters
- Shape:
输入: 或
输出: 或 , 其中