diff --git a/src/util/mod.rs b/src/util/mod.rs index 4846cd15..7ee995f5 100644 --- a/src/util/mod.rs +++ b/src/util/mod.rs @@ -35,17 +35,13 @@ pub(crate) fn transpose_poly_values(polys: Vec>) - transpose(&poly_values) } -pub(crate) fn transpose(matrix: &[Vec]) -> Vec> { - if matrix.is_empty() { - return Vec::new(); - } - - let old_rows = matrix.len(); - let old_cols = matrix[0].len(); - let mut transposed = vec![Vec::with_capacity(old_rows); old_cols]; - for new_r in 0..old_cols { - for old_row in matrix.iter() { - transposed[new_r].push(old_row[new_r].clone()); +pub(crate) fn transpose(matrix: &[Vec]) -> Vec> { + let l = matrix.len(); + let w = matrix[0].len(); + let mut transposed = vec![vec![F::ZERO; l]; w]; + for i in 0..w { + for j in 0..l { + transposed[i][j] = matrix[j][i]; } } transposed