Method __init__

src/sharp/models/blocks.py:60–64

Initialize ResidualBlock.

(self, residual: nn.Module, shortcut: nn.Module | None = None)

    """

    def __init__(self, residual: nn.Module, shortcut: nn.Module | None = None) -> None:
        """Initialize ResidualBlock."""
        super().__init__()
        self.residual = residual
        self.shortcut = shortcut

    def forward(self, x: torch.Tensor) -> torch.Tensor:
        """Apply residual block."""