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

rearrange

Browse Source
This commit is contained in:
conrad
2025-08-04 23:55:51 +10:00
parent 7eca200236
commit 567d6ccd2c
2 changed files with 27 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/armadillo_bits/op_mean_bones.hpp
+2 -2
View File
@@ -49,10 +49,10 @@ struct op_mean
inline static eT direct_mean(const eT* X_mem, const uword N);
template<typename eT>
inline static eT direct_mean_promote(const eT* X_mem, const uword N);
inline static eT direct_mean_robust(const eT old_mean, const eT* X_mem, const uword N);
template<typename eT>
inline static eT direct_mean_robust(const eT old_mean, const eT* X_mem, const uword N);
inline static eT direct_mean_promote(const eT* X_mem, const uword N);
template<typename eT>
inline static eT direct_mean_robust_promote(const eT old_mean, const eT* X_mem, const uword N);
include/armadillo_bits/op_mean_meat.hpp
+25 -25
View File
@@ -325,6 +325,31 @@ op_mean::direct_mean(const eT* X_mem, const uword N)
template<typename eT>
inline
eT
op_mean::direct_mean_robust(const eT old_mean, const eT* X_mem, const uword N)
{
arma_debug_sigprint();
// use an adapted form of the mean finding algorithm from the running_stat class
typedef typename get_pod_type<eT>::result T;
if(arrayops::is_finite(X_mem, N) == false) { return old_mean; }
eT r_mean = eT(0);
for(uword i=0; i < N; ++i)
{
r_mean = r_mean + (X_mem[i] - r_mean) / T(i+1);
}
return r_mean;
}
template<typename eT>
inline
eT
@@ -352,31 +377,6 @@ op_mean::direct_mean_promote(const eT* X_mem, const uword N)
template<typename eT>
inline
eT
op_mean::direct_mean_robust(const eT old_mean, const eT* X_mem, const uword N)
{
arma_debug_sigprint();
// use an adapted form of the mean finding algorithm from the running_stat class
typedef typename get_pod_type<eT>::result T;
if(arrayops::is_finite(X_mem, N) == false) { return old_mean; }
eT r_mean = eT(0);
for(uword i=0; i < N; ++i)
{
r_mean = r_mean + (X_mem[i] - r_mean) / T(i+1);
}
return r_mean;
}
template<typename eT>
inline
eT