Module flatten

Module flatten 

Source

Functionsยง

flatten
Flattens a contiguous range of dims into a tensor.