Tell Pytorch's Autograd not to include a operation for gradient computation

Question

I have a custom layer. Let the layer be called \'Gaussian\'

class Gaussian(nn.Module):
    ....
    def forward(self, x):
        x = x ** 2
        # Few oth