text_recognizer/networks/transformer/residual.py


1
2
3
4
5
6
7
8
9
10

"""Residual function."""
from torch import nn, Tensor


class Residual(nn.Module):
    """Residual block."""

    def forward(self, x: Tensor, residual: Tensor) -> Tensor:
        """Applies the residual function."""
        return x + residual