def forward()

in flow_layers/container.py [0:0]

• 14 lines of code
• 6 McCabe index (conditional complexity)

    def forward(self, x, logpx=None, reverse=False, inds=None, **kwargs):
        if inds is None:
            if reverse:
                inds = range(len(self.chain) - 1, -1, -1)
            else:
                inds = range(len(self.chain))

        if logpx is None:
            for i in inds:
                x = self.chain[i](x, reverse=reverse, **kwargs)
            return x
        else:
            for i in inds:
                x, logpx = self.chain[i](x, logpx, reverse=reverse, **kwargs)
            return x, logpx