You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.nn.Conv1dTranspose.rst 4.6 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. mindspore.nn.Conv1dTranspose
  2. =============================
  3. .. py:class:: mindspore.nn.Conv1dTranspose(in_channels, out_channels, kernel_size, stride=1, pad_mode='same', padding=0, dilation=1, group=1, has_bias=False, weight_init='normal', bias_init='zeros')
  4. 一维转置卷积层。
  5. 计算一维转置卷积,也称为反卷积(实际不是真正的反卷积)。
  6. 该算子可以看成Conv1d相对于其输入的梯度。
  7. `x` 的shape通常是 :math:`(N, C, W)` ,其中 :math:`N` 是batch size, :math:`C` 是通道数, :math:`W` 是特征长度。
  8. 对输入填充 :math:`dilation * (kernel\_size - 1) - padding` 个零。
  9. 因此,当Conv1d和ConvTranspose1d使用相同的参数初始化时,它们的输入和输出shape是互逆的。
  10. 但是,当stride>1时,Conv1d将多个输入的shape映射到同一个输出shape。
  11. 输出宽度定义如下:
  12. .. math::
  13. W_{out} = \begin{cases}
  14. (W_{in} - 1) \times \text{stride} - 2 \times \text{padding} + \text{dilation} \times
  15. (\text{ks_w} - 1) + 1, & \text{if pad_mode='pad'}\\
  16. (W_{in} + \text{stride} - 1)/\text{stride}, & \text{if pad_mode='same'}\\
  17. (W_{in} - 1) \times \text{stride} + \text{dilation} \times
  18. (\text{ks_w} - 1) + 1, & \text{if pad_mode='valid'}
  19. \end{cases}
  20. 其中 :math:`\text{ks_w}` 是卷积核的宽度。
  21. **参数:**
  22. - **in_channels** (int) - 输入的通道数。
  23. - **out_channels** (int) - 输出的通道数。
  24. - **kernel_size** (int) - 指定一维卷积窗口的宽度。
  25. - **stride** (int) - 步长大小,表示移到宽度。默认值:1。
  26. - **pad_mode** (str) - 选择填充模式。可选值为"pad","same","valid"。默认值:"same"。
  27. - same:采用补全方式。输出的宽度与输入 `x` 一致。填充总数将在水平上进行计算。并尽可能均匀分布到左侧和右侧。否则,最后一次将从底部和右侧进行额外的填充。若设置该模式, `padding` 必须为0。
  28. - valid:采用的丢弃方式。在不填充的前提下返回可能大的宽度的输出。多余的像素会被丢弃。如果设置此模式,则 `padding` 必须为0。
  29. - pad:输入 `x` 两侧的隐式填充。 `padding` 的数量将填充到输入Tensor边框上。 `padding` 必须大于或等于0。
  30. - **padding** (int) - 输入`x`两侧的隐式填充。默认值:0。
  31. - **dilation** (int) - 指定用于扩张卷积的扩张速率。如果设置为 :math:`k > 1` ,则每个采样位置都跳过 :math:`k - 1` 个像素。其值必须大于或等于1,并以输入 `x` 的宽度为界。默认值:1。
  32. - **group** (int) - 将过滤器拆分为组, `in_ channels` 和 `out_channels` 必须可被组数整除。当组数>1时,不支持Davinci设备。默认值:1。
  33. - **has_bias** (bool) - 指定图层是否使用偏置矢量。默认值:False。
  34. - **weight_init** (`Union[Tensor, str, Initializer, numbers.Number]`) – 卷积核的初始化方法。它可以是Tensor,str,初始化实例或numbers.Number。当使用str时,可选“TruncatedNormal”,“Normal”,“Uniform”,“HeUniform”和“XavierUniform”分布以及常量“One”和“Zero”分布的值,可接受别名“ xavier_uniform”,“ he_uniform”,“ ones”和“ zeros”。上述字符串大小写均可。更多细节请参考Initializer的值。默认值:“normal”。
  35. - **bias_init** (`Union[Tensor, str, Initializer, numbers.Number]`) – 偏置向量的初始化方法。可以使用的初始化方法和字符串与“weight_init”相同。更多细节请参考Initializer的值。默认值:“zeros”。
  36. **输入:**
  37. - **x** (Tensor) - shape为 :math:`(N, C_{in}, W_{in})` 的Tensor。
  38. **输出:**
  39. Tensor,shape为 :math:`(N, C_{out}, W_{out})` 。
  40. **异常:**
  41. - **TypeError** - `in_channels` 、 `out_channels` 、 `kernel_size` 、 `stride` 、 `padding`或 `dilation` 不是int。
  42. - **ValueError** - `in_channels` 、 `out_channels` 、 `kernel_size` 、 `stride` 或 `dilation` 小于1。
  43. - **ValueError** - `padding` 小于0。
  44. - **ValueError** - `pad_mode` 不是'same','valid',或'pad'。
  45. **支持平台:**
  46. ``Ascend`` ``GPU`` ``CPU``
  47. **样例:**
  48. >>> net = nn.Conv1dTranspose(3, 64, 4, has_bias=False, weight_init='normal', pad_mode='pad')
  49. >>> x = Tensor(np.ones([1, 3, 50]), mindspore.float32)
  50. >>> output = net(x).shape
  51. >>> print(output)
  52. (1, 64, 53)