in flowtorch/bijectors/spline.py [0:0]
def __init__(
self,
params_fn: Optional[flowtorch.Lazy] = None,
*,
shape: torch.Size,
context_shape: Optional[torch.Size] = None,
count_bins: int = 8,
bound: float = 3.0,
order: str = "linear"