:py:mod:`anomalib.models.components.layers` =========================================== .. py:module:: anomalib.models.components.layers .. autoapi-nested-parse:: Neural network layers. Submodules ---------- .. toctree:: :titlesonly: :maxdepth: 1 sspcab/index.rst Package Contents ---------------- Classes ~~~~~~~ .. autoapisummary:: anomalib.models.components.layers.SSPCAB .. py:class:: SSPCAB(in_channels: int, kernel_size: int = 1, dilation: int = 1, reduction_ratio: int = 8) Bases: :py:obj:`torch.nn.Module` SSPCAB block. :param in_channels: Number of input channels. :type in_channels: int :param kernel_size: Size of the receptive fields of the masked convolution kernel. :type kernel_size: int :param dilation: Dilation factor of the masked convolution kernel. :type dilation: int :param reduction_ratio: Reduction ratio of the attention module. :type reduction_ratio: int .. py:method:: forward(inputs: torch.Tensor) -> torch.Tensor Forward pass through the SSPCAB block.