flatten

Also written: Flatten, nn.Flatten, .flatten()

Turns each example's multi-dimensional feature maps into one long row. `nn.Flatten()` turns (32, 7, 7) into 1,568 numbers before the linear layers.