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

specialisations for randu() and randn()

Browse Source
This commit is contained in:
conrad
2026-04-04 19:47:20 +10:00
parent 43b071020a
commit 7e4a956dbd
2 changed files with 90 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/armadillo_bits/subview_bones.hpp
+4
View File
@@ -400,6 +400,8 @@ class subview_col : public subview<eT>
inline void fill(const eT val);
inline void zeros();
inline void ones();
inline void randu();
inline void randn();
arma_warn_unused inline bool is_finite() const;
arma_warn_unused inline bool is_zero(const pod_type tol = 0) const;
@@ -563,6 +565,8 @@ class subview_row : public subview<eT>
inline void fill(const eT val);
inline void zeros();
inline void ones();
inline void randu();
inline void randn();
arma_warn_unused inline bool is_finite() const;
arma_warn_unused inline bool is_zero(const pod_type tol = 0) const;
include/armadillo_bits/subview_meat.hpp
+86 -36
View File
@@ -1151,30 +1151,17 @@ subview<eT>::randu()
const uword s_n_rows = s.n_rows;
const uword s_n_cols = s.n_cols;
if( (s_n_rows == 0) || (s_n_cols == 0) ) { return; }
if(s_n_rows == 0) { return; }
if(s_n_rows == 1)
if( (s.aux_row1 == 0) && (s_n_rows == s.m.n_rows) )
{
podarray<eT> tmp(s_n_cols);
eT* tmp_mem = tmp.memptr();
arma_rng::randu<eT>::fill( tmp_mem, s_n_cols );
for(uword ii=0; ii < s_n_cols; ++ii) { at(0,ii) = tmp_mem[ii]; }
arma_rng::randu<eT>::fill( s.colptr(0), s.n_elem );
}
else
{
if( (s.aux_row1 == 0) && (s_n_rows == s.m.n_rows) )
for(uword ii=0; ii < s_n_cols; ++ii)
{
arma_rng::randu<eT>::fill( s.colptr(0), s.n_elem );
}
else
{
for(uword ii=0; ii < s_n_cols; ++ii)
{
arma_rng::randu<eT>::fill( s.colptr(ii), s_n_rows );
}
arma_rng::randu<eT>::fill( s.colptr(ii), s_n_rows );
}
}
}
@@ -1193,30 +1180,17 @@ subview<eT>::randn()
const uword s_n_rows = s.n_rows;
const uword s_n_cols = s.n_cols;
if( (s_n_rows == 0) || (s_n_cols == 0) ) { return; }
if(s_n_rows == 0) { return; }
if(s_n_rows == 1)
if( (s.aux_row1 == 0) && (s_n_rows == s.m.n_rows) )
{
podarray<eT> tmp(s_n_cols);
eT* tmp_mem = tmp.memptr();
arma_rng::randn<eT>::fill( tmp_mem, s_n_cols );
for(uword ii=0; ii < s_n_cols; ++ii) { at(0,ii) = tmp_mem[ii]; }
arma_rng::randn<eT>::fill( s.colptr(0), s.n_elem );
}
else
{
if( (s.aux_row1 == 0) && (s_n_rows == s.m.n_rows) )
for(uword ii=0; ii < s_n_cols; ++ii)
{
arma_rng::randn<eT>::fill( s.colptr(0), s.n_elem );
}
else
{
for(uword ii=0; ii < s_n_cols; ++ii)
{
arma_rng::randn<eT>::fill( s.colptr(ii), s_n_rows );
}
arma_rng::randn<eT>::fill( s.colptr(ii), s_n_rows );
}
}
}
@@ -3561,6 +3535,30 @@ subview_col<eT>::ones()
template<typename eT>
inline
void
subview_col<eT>::randu()
{
arma_debug_sigprint();
arma_rng::randu<eT>::fill( access::rwp(colmem), subview<eT>::n_rows );
}
template<typename eT>
inline
void
subview_col<eT>::randn()
{
arma_debug_sigprint();
arma_rng::randn<eT>::fill( access::rwp(colmem), subview<eT>::n_rows );
}
template<typename eT>
inline
bool
@@ -4650,6 +4648,58 @@ subview_row<eT>::ones()
template<typename eT>
inline
void
subview_row<eT>::randu()
{
arma_debug_sigprint();
const uword local_s_n_cols = subview<eT>::n_cols;
const uword local_m_n_rows = subview<eT>::m.n_rows;
podarray<eT> tmp(local_s_n_cols);
eT* tmp_mem = tmp.memptr();
arma_rng::randu<eT>::fill( tmp_mem, local_s_n_cols );
eT* mem_ptr = access::rwp(rowmem);
for(uword ii=0; ii < local_s_n_cols; ++ii)
{
(*mem_ptr) = tmp_mem[ii]; mem_ptr += local_m_n_rows;
}
}
template<typename eT>
inline
void
subview_row<eT>::randn()
{
arma_debug_sigprint();
const uword local_s_n_cols = subview<eT>::n_cols;
const uword local_m_n_rows = subview<eT>::m.n_rows;
podarray<eT> tmp(local_s_n_cols);
eT* tmp_mem = tmp.memptr();
arma_rng::randn<eT>::fill( tmp_mem, local_s_n_cols );
eT* mem_ptr = access::rwp(rowmem);
for(uword ii=0; ii < local_s_n_cols; ++ii)
{
(*mem_ptr) = tmp_mem[ii]; mem_ptr += local_m_n_rows;
}
}
template<typename eT>
inline
bool