Function rai_core::ops::flatten

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