beanmachine.ppl.compiler.sizer module

class beanmachine.ppl.compiler.sizer.Sizer: Bases: beanmachine.ppl.compiler.typer_base.TyperBase[torch.Size]

beanmachine.ppl.compiler.sizer.is_scalar(s: torch.Size) → bool

beanmachine.ppl.compiler.sizer.size_to_str(size: torch.Size) → str