Function batch_matrix 

pub fn batch_matrix(
    element_type: impl Into<String>,
    batch: DimExpr,
    rows: DimExpr,
    cols: DimExpr,
) -> DependentType

Batch of matrices.