def inv_permutation()

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

• 8 lines of code
• 2 McCabe index (conditional complexity)

    def inv_permutation(self) -> Optional[torch.Tensor]:
        if self.permutation is None:
            return None

        result = torch.empty_like(self.permutation, dtype=torch.long)
        result[self.permutation] = torch.arange(
            self.permutation.size(0), dtype=torch.long, device=self.permutation.device
        )
        return result