def __init__()

in flowtorch/bijectors/permute.py [0:0]

• 7 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(
        self,
        params_fn: Optional[flowtorch.Lazy] = None,
        *,
        shape: torch.Size,
        context_shape: Optional[torch.Size] = None,
        permutation: Optional[torch.Tensor] = None