probly.layers.torch.pack_interval

probly.layers.torch.pack_interval(x: Tensor, channel_dim: int = 1) → Tensor

Promote x to a packed interval tensor with lo == hi == x.

Doubles x along channel_dim. Call once at the network entry, before the first interval layer. Analogous to BatchEnsemble’s tile_inputs (which doubles the batch dim instead).