Conv2dNormActivation

class torchvision.ops.Conv2dNormActivation(in_channels: int, out_channels: int, kernel_size: ~typing.Union[int, ~typing.Tuple[int, int]] = 3, stride: ~typing.Union[int, ~typing.Tuple[int, int]] = 1, padding: ~typing.Optional[~typing.Union[int, ~typing.Tuple[int, int], str]] = None, groups: int = 1, norm_layer: ~typing.Optional[~typing.Callable[[...], ~torch.nn.modules.module.Module]] = <class 'torch.nn.modules.batchnorm.BatchNorm2d'>, activation_layer: ~typing.Optional[~typing.Callable[[...], ~torch.nn.modules.module.Module]] = <class 'torch.nn.modules.activation.ReLU'>, dilation: ~typing.Union[int, ~typing.Tuple[int, int]] = 1, inplace: ~typing.Optional[bool] = True, bias: ~typing.Optional[bool] = None)

用於 Convolution2d-Normalization-Activation 區塊的可配置區塊。

參數:

• in_channels (int) – 輸入影像中的通道數

• out_channels (int) – Convolution-Normalization-Activation 區塊產生的通道數

• kernel_size – (int, optional): 卷積核的大小。預設值：3

• stride (int, optional) – 卷積的步幅。預設值：1

• padding (int, tuple 或 str, optional) – 新增到輸入四個邊的填充。預設值：None，在這種情況下，它將計算為 padding = (kernel_size - 1) // 2 * dilation

• groups (int, optional) – 從輸入通道到輸出通道的封鎖連線數。預設值：1

• norm_layer (Callable[..., torch.nn.Module], optional) – 將堆疊在卷積層之上的 Norm 層。如果 None，則不會使用此層。預設值：torch.nn.BatchNorm2d

• activation_layer (Callable[..., torch.nn.Module], optional) – 將堆疊在標準化層（如果不是 None）之上，否則堆疊在卷積層之上的啟動函數。如果 None，則不會使用此層。預設值：torch.nn.ReLU

• dilation (int) – 卷積核元素之間的間距。預設值：1

• inplace (bool) – 啟動層的參數，可以選擇就地執行操作。預設值 True

• bias (bool, optional) – 是否在卷積層中使用偏差。預設情況下，如果 norm_layer is None，則包含偏差。