PyTorch Module

class MyModule(nn.Module):
  """Some Information about MyModule"""
  def __init__(self):
    super(MyModule, self).__init__()

  def forward(self, x):

    return x