import torch

class none_class(torch.nn.Module):
    def forward(self, x):
        return x