Function rai_core::ops::chunk

source · 
pub fn chunk(x: &Tensor, chunks: usize, dim: impl Dim) -> Vec<Tensor>