Chris@49
|
1 // Copyright (C) 2012 Ryan Curtin
|
Chris@49
|
2 // Copyright (C) 2012 Conrad Sanderson
|
Chris@49
|
3 //
|
Chris@49
|
4 // This Source Code Form is subject to the terms of the Mozilla Public
|
Chris@49
|
5 // License, v. 2.0. If a copy of the MPL was not distributed with this
|
Chris@49
|
6 // file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
Chris@49
|
7
|
Chris@49
|
8
|
Chris@49
|
9 //! \addtogroup spop_var
|
Chris@49
|
10 //! @{
|
Chris@49
|
11
|
Chris@49
|
12
|
Chris@49
|
13
|
Chris@49
|
14 //! Class for finding variance values of a sparse matrix
|
Chris@49
|
15 class spop_var
|
Chris@49
|
16 {
|
Chris@49
|
17 public:
|
Chris@49
|
18
|
Chris@49
|
19 template<typename T1>
|
Chris@49
|
20 inline static void apply(SpMat<typename T1::pod_type>& out, const mtSpOp<typename T1::pod_type, T1, spop_var>& in);
|
Chris@49
|
21
|
Chris@49
|
22 template<typename T1>
|
Chris@49
|
23 inline static void apply_noalias(SpMat<typename T1::pod_type>& out, const SpProxy<T1>& p, const uword norm_type, const uword dim);
|
Chris@49
|
24
|
Chris@49
|
25 // Calculate variance of a sparse vector, where we can directly use the memory.
|
Chris@49
|
26 template<typename T1>
|
Chris@49
|
27 inline static typename T1::pod_type var_vec(const T1& X, const uword norm_type = 0);
|
Chris@49
|
28
|
Chris@49
|
29 // Calculate the variance directly. Because this is for sparse matrices, we
|
Chris@49
|
30 // specify both the number of elements in the array (the length of the array)
|
Chris@49
|
31 // as well as the actual number of elements when zeros are included.
|
Chris@49
|
32 template<typename eT>
|
Chris@49
|
33 inline static eT direct_var(const eT* const X, const uword length, const uword N, const uword norm_type = 0);
|
Chris@49
|
34
|
Chris@49
|
35 // For complex numbers.
|
Chris@49
|
36
|
Chris@49
|
37 template<typename T>
|
Chris@49
|
38 inline static T direct_var(const std::complex<T>* const X, const uword length, const uword N, const uword norm_type = 0);
|
Chris@49
|
39
|
Chris@49
|
40 // Calculate the variance using iterators, for non-complex numbers.
|
Chris@49
|
41 template<typename T1, typename eT>
|
Chris@49
|
42 inline static eT iterator_var(T1& it, const T1& end, const uword n_zero, const uword norm_type, const eT junk1, const typename arma_not_cx<eT>::result* junk2 = 0);
|
Chris@49
|
43
|
Chris@49
|
44 // Calculate the variance using iterators, for complex numbers.
|
Chris@49
|
45 template<typename T1, typename eT>
|
Chris@49
|
46 inline static typename get_pod_type<eT>::result iterator_var(T1& it, const T1& end, const uword n_zero, const uword norm_type, const eT junk1, const typename arma_cx_only<eT>::result* junk2 = 0);
|
Chris@49
|
47
|
Chris@49
|
48 };
|
Chris@49
|
49
|
Chris@49
|
50
|
Chris@49
|
51
|
Chris@49
|
52 //! @}
|
Chris@49
|
53
|