Module flatten

Module flatten 

Source

Structsยง

Flatten
Flatten layer: collapses dimensions [start_dim..=end_dim] into one.