85d3fc00cd039f279463307e70ce645f982b32d2,pyro/nn/dense_nn.py,DenseNN,forward,#DenseNN#Any#,63

Before Change

        
        The forward method
        
        h = x
        for layer in self.layers[:-1]:
            h = self.f(layer(h))
        h = self.layers[-1](h)

        // Shape the output, squeezing the parameter dimension if all ones
        if self.output_multiplier == 1:
            return h
        else:
            h = h.reshape(list(x.size()[:-1]) + [self.output_multiplier])

            if self.count_params == 1:

After Change

        
        The forward method
        
        return self._forward(x)

Instances