class Transformer:
    def __init__(self, d_model: int):
        self.d_model = d_model

    def forward(self, x):
        return x