Chris@49
|
1 // Copyright (C) 2009-2012 NICTA (www.nicta.com.au)
|
Chris@49
|
2 // Copyright (C) 2009-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 fn_prod
|
Chris@49
|
10 //! @{
|
Chris@49
|
11
|
Chris@49
|
12
|
Chris@49
|
13 //! \brief
|
Chris@49
|
14 //! Delayed product of elements of a matrix along a specified dimension (either rows or columns).
|
Chris@49
|
15 //! The result is stored in a dense matrix that has either one column or one row.
|
Chris@49
|
16 //! For dim = 0, find the sum of each column (i.e. traverse across rows)
|
Chris@49
|
17 //! For dim = 1, find the sum of each row (i.e. traverse across columns)
|
Chris@49
|
18 //! The default is dim = 0.
|
Chris@49
|
19 //! NOTE: this function works differently than in Matlab/Octave.
|
Chris@49
|
20
|
Chris@49
|
21 template<typename T1>
|
Chris@49
|
22 arma_inline
|
Chris@49
|
23 const Op<T1, op_prod>
|
Chris@49
|
24 prod
|
Chris@49
|
25 (
|
Chris@49
|
26 const T1& X,
|
Chris@49
|
27 const uword dim = 0,
|
Chris@49
|
28 const typename enable_if< is_arma_type<T1>::value == true >::result* junk1 = 0,
|
Chris@49
|
29 const typename enable_if< resolves_to_vector<T1>::value == false >::result* junk2 = 0
|
Chris@49
|
30 )
|
Chris@49
|
31 {
|
Chris@49
|
32 arma_extra_debug_sigprint();
|
Chris@49
|
33 arma_ignore(junk1);
|
Chris@49
|
34 arma_ignore(junk2);
|
Chris@49
|
35
|
Chris@49
|
36 return Op<T1, op_prod>(X, dim, 0);
|
Chris@49
|
37 }
|
Chris@49
|
38
|
Chris@49
|
39
|
Chris@49
|
40
|
Chris@49
|
41 template<typename T1>
|
Chris@49
|
42 arma_inline
|
Chris@49
|
43 const Op<T1, op_prod>
|
Chris@49
|
44 prod
|
Chris@49
|
45 (
|
Chris@49
|
46 const T1& X,
|
Chris@49
|
47 const uword dim,
|
Chris@49
|
48 const typename enable_if<resolves_to_vector<T1>::value == true>::result* junk = 0
|
Chris@49
|
49 )
|
Chris@49
|
50 {
|
Chris@49
|
51 arma_extra_debug_sigprint();
|
Chris@49
|
52 arma_ignore(junk);
|
Chris@49
|
53
|
Chris@49
|
54 return Op<T1, op_prod>(X, dim, 0);
|
Chris@49
|
55 }
|
Chris@49
|
56
|
Chris@49
|
57
|
Chris@49
|
58
|
Chris@49
|
59 template<typename T1>
|
Chris@49
|
60 inline
|
Chris@49
|
61 arma_warn_unused
|
Chris@49
|
62 typename T1::elem_type
|
Chris@49
|
63 prod
|
Chris@49
|
64 (
|
Chris@49
|
65 const T1& X,
|
Chris@49
|
66 const arma_empty_class junk1 = arma_empty_class(),
|
Chris@49
|
67 const typename enable_if<resolves_to_vector<T1>::value == true>::result* junk2 = 0
|
Chris@49
|
68 )
|
Chris@49
|
69 {
|
Chris@49
|
70 arma_extra_debug_sigprint();
|
Chris@49
|
71 arma_ignore(junk1);
|
Chris@49
|
72 arma_ignore(junk2);
|
Chris@49
|
73
|
Chris@49
|
74 return op_prod::prod( X );
|
Chris@49
|
75 }
|
Chris@49
|
76
|
Chris@49
|
77
|
Chris@49
|
78
|
Chris@49
|
79 //! \brief
|
Chris@49
|
80 //! Immediate 'product of all values' operation,
|
Chris@49
|
81 //! invoked, for example, by: prod(prod(A))
|
Chris@49
|
82
|
Chris@49
|
83 template<typename T1>
|
Chris@49
|
84 inline
|
Chris@49
|
85 arma_warn_unused
|
Chris@49
|
86 typename T1::elem_type
|
Chris@49
|
87 prod(const Op<T1, op_prod>& in)
|
Chris@49
|
88 {
|
Chris@49
|
89 arma_extra_debug_sigprint();
|
Chris@49
|
90 arma_extra_debug_print("prod(): two consecutive prod() calls detected");
|
Chris@49
|
91
|
Chris@49
|
92 return op_prod::prod( in.m );
|
Chris@49
|
93 }
|
Chris@49
|
94
|
Chris@49
|
95
|
Chris@49
|
96
|
Chris@49
|
97 template<typename T1>
|
Chris@49
|
98 inline
|
Chris@49
|
99 const Op<Op<T1, op_prod>, op_prod>
|
Chris@49
|
100 prod(const Op<T1, op_prod>& in, const uword dim)
|
Chris@49
|
101 {
|
Chris@49
|
102 arma_extra_debug_sigprint();
|
Chris@49
|
103
|
Chris@49
|
104 return Op<Op<T1, op_prod>, op_prod>(in, dim, 0);
|
Chris@49
|
105 }
|
Chris@49
|
106
|
Chris@49
|
107
|
Chris@49
|
108
|
Chris@49
|
109 template<typename T>
|
Chris@49
|
110 arma_inline
|
Chris@49
|
111 arma_warn_unused
|
Chris@49
|
112 const typename arma_scalar_only<T>::result &
|
Chris@49
|
113 prod(const T& x)
|
Chris@49
|
114 {
|
Chris@49
|
115 return x;
|
Chris@49
|
116 }
|
Chris@49
|
117
|
Chris@49
|
118
|
Chris@49
|
119
|
Chris@49
|
120 //! @}
|