mlx.nn.ConvTranspose1d

mlx.nn.ConvTranspose1d#

class ConvTranspose1d(in_channels: int, out_channels: int, kernel_size: int, stride: int = 1, padding: int = 0, dilation: int = 1, output_padding: int = 0, bias: bool = True)#

在多通道输入序列上应用一维反卷积。

通道维度预期在最后，即输入形状应为 NLC，其中

  • N 是批量维度

  • L 是序列长度

  • C 是输入通道数

参数：

  • in_channels (int) – 输入通道数

  • out_channels (int) – 输出通道数

  • kernel_size (int) – 卷积核大小

  • stride (int, optional) – 应用滤波器时的步幅。默认值：1。

  • padding (int, optional) – 输入的零填充位数。默认值：0。

  • dilation (int, optional) – 卷积的膨胀率。

  • output_padding (int, optional) – 添加到输出形状一侧的额外大小。默认值：0。

  • bias (bool, optional) – 如果为 True，则向输出添加可学习的偏置。默认值：True

方法

目录