cubecl_linalg::tensor

Function matrix_batch_layout

Source 
pub fn matrix_batch_layout(strides: &[usize]) -> MatrixBatchLayout
Expand description

Return the layout of a matrix batch given the strides.