Function flatten

Flattens a contiguous range of dims into a 1D tensor.

flatten, unlike other shape operators, returns the input tensor on a no-op (unless a 0D tensor is flattened, in which case it's returned in 1D).