phynexis-deps/armadillo-code
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

optimisations for common cases

Browse Source
This commit is contained in:
conrad
2025-01-17 13:33:13 +10:00
parent 64de62450d
commit 32fd3716fe
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/armadillo_bits/op_reshape_meat.hpp
+5 -1
View File
@@ -83,7 +83,11 @@ op_reshape::apply_mat_inplace(Mat<eT>& A, const uword new_n_rows, const uword ne
if(A.is_empty()) { A.zeros(new_n_rows, new_n_cols); return; }
if( (A.n_rows == new_n_cols) && (A.n_cols == new_n_rows) ) { A.set_size(new_n_rows, new_n_cols); return; }
const bool is_into_colvec = ( (new_n_cols == uword(1)) && (new_n_rows == A.n_elem) );
const bool is_into_rowvec = ( (new_n_rows == uword(1)) && (new_n_cols == A.n_elem) );
const bool is_rowcol_swap = ( (A.n_rows == new_n_cols) && (A.n_cols == new_n_rows) );
if(is_into_colvec || is_into_rowvec || is_rowcol_swap) { A.set_size(new_n_rows, new_n_cols); return; }
Mat<eT> B(new_n_rows, new_n_cols, arma_nozeros_indicator());