ConvTranspose2d

class mmcv.cnn.ConvTranspose2d(in_channels: int, out_channels: int, kernel_size: int | tuple[int, int], stride: int | tuple[int, int] = 1, padding: int | tuple[int, int] = 0, output_padding: int | tuple[int, int] = 0, groups: int = 1, bias: bool = True, dilation: int | tuple[int, int] = 1, padding_mode: Literal['zeros', 'reflect', 'replicate', 'circular'] = 'zeros', device=None, dtype=None)

forward(x: Tensor) → Tensor

Performs the forward pass.

input

The input tensor.

Type:

Tensor

output_size

A list of integers representing the size of the output tensor. Default is None.

Type:

list[int], optional