Function flatten

pub fn flatten<T: FlattenArgs>(x: &Tensor, args: T) -> Tensor