def register()

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

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

    def register(
        self,
        input: Tensor,
        output: Tensor,
        context: Optional[Tensor],
        bijector: "Bijector",
        log_detJ: Optional[Tensor],
        mode: str,