in flowtorch/bijectors/permute.py [0:0]
def _forward( self, x: torch.Tensor, params: Optional[Sequence[torch.Tensor]]