`Flatten`

Collapse every axis after the first into one.

class Flatten : Module— python/vkml/nn.py:581

A view, not a copy — it reshapes, so no data moves. Typically the join between a convolutional stack and a linear head.

Construction¶

`init`¶

def __init__self, start_dim: int=1— python/vkml/nn.py:588

Forward¶

`forward`¶

def forwardself, x: V.Tensor -> V.Tensor— python/vkml/nn.py:592

Other members¶

`repr`¶

def __repr__self -> str— python/vkml/nn.py:600

See also Sequential, Linear