Skip to main content

fuse_dims

Function fuse_dims 

Source
pub fn fuse_dims(dims: &[usize], all_strides: &[&[isize]]) -> Vec<usize>
Expand description

Fuse contiguous dimensions across multiple arrays.

This function fuses subsequent dimensions that are contiguous in memory for all arrays. If strides[k][i] == dims[i-1] * strides[k][i-1] for all k, dimensions i-1 and i can be merged.