def METHOD_NAME(in_channels, out_channels, kernel_size, stride, padding,
         dilation):
    return nn.Conv2d(
        in_channels=in_channels,
        out_channels=out_channels,
        kernel_size=kernel_size,
        stride=stride,
        padding=padding,
        dilation=dilation)