conv_transpose2d

dragon.vm.torch.nn.functional.conv_transpose2d(
  input,
  weight,
  bias=None,
  stride=1,
  padding=0,
  output_padding=0,
  groups=1,
  dilation=1
)[source]

Apply the 2d deconvolution to input.

Parameters:
  • input (dragon.vm.torch.Tensor) The input tensor.
  • weight (dragon.vm.torch.Tensor) The weight tensor.
  • bias (dragon.vm.torch.Tensor, optional) The bias tensor.
  • stride (Union[int, Sequence[int]], optional, default=1) The stride of convolution window.
  • padding (Union[int, Sequence[int]], optional, default=0) The zero padding size.
  • output_padding (int, optional, default=1) The additional size added to the output shape.
  • groups (int, optional, default=1) The number of groups to split input channels.
  • dilation (Union[int, Sequence[int]], optional, default=1) The rate of dilated kernel.
Returns:

dragon.vm.torch.Tensor The output tensor.