matrix_batch_layout

cubecl_std::tensor

Function matrix_batch_layout 

Source 
pub fn matrix_batch_layout(
    strides: &[usize],
    scheme: Option<&QuantScheme>,
) -> MatrixBatchLayout
Expand description

Return the layout of a matrix batch given the strides.