in flowtorch/bijectors/elementwise.py [0:0]
def __init__( self, params_fn: Optional[flowtorch.Lazy] = None, *, shape: torch.Size, context_shape: Optional[torch.Size] = None, **kwargs: Any