Mojo function mat_identity fn def mat_identity(rows: Int, cols: Int) -> Mat Args: rows (Int) cols (Int) Returns: Mat