def METHOD_NAME(self, x: torch.Tensor) -> torch.Tensor:
    """Defines the computation performed at every call.
    Args:
        x (torch.Tensor): The input data.
    Returns:
        torch.Tensor: The output of the module.
    """
    x = self.conv_s(x)
    x = self.bn_s(x)
    x = self.relu(x)
    x = self.conv_t(x)
    return x