import torch
import torch.nn as nn
     

class MyModel(nn.Module):
    def __init__(self, in_features, out_features):

        super(MyModel, self).__init__()

        pass

    def forward(self, x):
        
        out = x

        return out
