Flatten

Flatten(start_dim=1, end_dim=-1)

Merges the inclusive dimension range [start_dim, end_dim] into one dimension; a view when the layout permits.

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

Args: start_dim: the first dimension to merge. end_dim: the last dimension to merge; negative counts from the end.

`init`

__init__(self, start_dim: 'int' = 1, end_dim: 'int' = -1) -> 'None'

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

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(self, input: 'Tensor') -> 'Tensor'

forward(input) -> Tensor

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