Now we have defined the Gaussian layer, I want to know how to use it in a network of several layers. For example,
class MyNet(nn.Module):
    def __init__(self):
        super(MyNet, self).__init__()
        self.fc1 = nn.Linear(100, 20)
        self.gauss = Gaussian()
    def forward(self, x):
        x = F.relu(self.fc1(x))
        x = self.gauss(x)

net = MyNet()

Is the above code correct? Or do we have to set up other things in order to use the newly-defined layer in other network?