Unflatten

Unflatten(dim, unflattened_size)

Splits the dimension ``dim`` into ``sizes``, whose product must equal its
extent; a view when the layout permits.

The module form of ``clika_runtime.unflatten``: construct it once with the
settings, then call it like a function.

Args:
    dim: the dimension to split; negative counts from the end.
    unflattened_size: the extents the dimension splits into.

__init__

__init__(self, dim: 'int', unflattened_size: 'int | Sequence[int]') -> 'None'

Initialize self. See help(type(self)) for accurate signature.

extra_repr

extra_repr(self) -> 'str'

extra_repr() -> str

One line of per-class detail for :meth:__repr__; a layer prints its geometry here (in_features=64, out_features=256).

forward

forward(self, input: 'Tensor') -> 'Tensor'

forward(input) -> Tensor

Runs clika_runtime.unflatten on input with the stored settings; returns a new tensor.