Chris@49
|
1 // Copyright (C) 2012 Conrad Sanderson
|
Chris@49
|
2 //
|
Chris@49
|
3 // This Source Code Form is subject to the terms of the Mozilla Public
|
Chris@49
|
4 // License, v. 2.0. If a copy of the MPL was not distributed with this
|
Chris@49
|
5 // file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
Chris@49
|
6
|
Chris@49
|
7
|
Chris@49
|
8 //! \addtogroup fn_sprandu
|
Chris@49
|
9 //! @{
|
Chris@49
|
10
|
Chris@49
|
11
|
Chris@49
|
12
|
Chris@49
|
13 //! Generate a sparse matrix with a randomly selected subset of the elements
|
Chris@49
|
14 //! set to random values in the [0,1] interval (uniform distribution)
|
Chris@49
|
15 template<typename obj_type>
|
Chris@49
|
16 inline
|
Chris@49
|
17 obj_type
|
Chris@49
|
18 sprandu
|
Chris@49
|
19 (
|
Chris@49
|
20 const uword n_rows,
|
Chris@49
|
21 const uword n_cols,
|
Chris@49
|
22 const double density,
|
Chris@49
|
23 const typename arma_SpMat_SpCol_SpRow_only<obj_type>::result* junk = 0
|
Chris@49
|
24 )
|
Chris@49
|
25 {
|
Chris@49
|
26 arma_extra_debug_sigprint();
|
Chris@49
|
27 arma_ignore(junk);
|
Chris@49
|
28
|
Chris@49
|
29 if(is_SpCol<obj_type>::value == true)
|
Chris@49
|
30 {
|
Chris@49
|
31 arma_debug_check( (n_cols != 1), "sprandu(): incompatible size" );
|
Chris@49
|
32 }
|
Chris@49
|
33 else
|
Chris@49
|
34 if(is_SpRow<obj_type>::value == true)
|
Chris@49
|
35 {
|
Chris@49
|
36 arma_debug_check( (n_rows != 1), "sprandu(): incompatible size" );
|
Chris@49
|
37 }
|
Chris@49
|
38
|
Chris@49
|
39 obj_type out;
|
Chris@49
|
40
|
Chris@49
|
41 out.sprandu(n_rows, n_cols, density);
|
Chris@49
|
42
|
Chris@49
|
43 return out;
|
Chris@49
|
44 }
|
Chris@49
|
45
|
Chris@49
|
46
|
Chris@49
|
47
|
Chris@49
|
48 inline
|
Chris@49
|
49 sp_mat
|
Chris@49
|
50 sprandu(const uword n_rows, const uword n_cols, const double density)
|
Chris@49
|
51 {
|
Chris@49
|
52 arma_extra_debug_sigprint();
|
Chris@49
|
53
|
Chris@49
|
54 sp_mat out;
|
Chris@49
|
55
|
Chris@49
|
56 out.sprandu(n_rows, n_cols, density);
|
Chris@49
|
57
|
Chris@49
|
58 return out;
|
Chris@49
|
59 }
|
Chris@49
|
60
|
Chris@49
|
61
|
Chris@49
|
62
|
Chris@49
|
63 //! Generate a sparse matrix with the non-zero values in the same locations as in the given sparse matrix X,
|
Chris@49
|
64 //! with the non-zero values set to random values in the [0,1] interval (uniform distribution)
|
Chris@49
|
65 template<typename T1>
|
Chris@49
|
66 inline
|
Chris@49
|
67 SpMat<typename T1::elem_type>
|
Chris@49
|
68 sprandu(const SpBase<typename T1::elem_type, T1>& X)
|
Chris@49
|
69 {
|
Chris@49
|
70 arma_extra_debug_sigprint();
|
Chris@49
|
71
|
Chris@49
|
72 typedef typename T1::elem_type eT;
|
Chris@49
|
73
|
Chris@49
|
74 SpMat<eT> out( X.get_ref() );
|
Chris@49
|
75
|
Chris@49
|
76 eop_aux_randu<eT>::fill( access::rwp(out.values), out.n_nonzero );
|
Chris@49
|
77
|
Chris@49
|
78 return out;
|
Chris@49
|
79 }
|
Chris@49
|
80
|
Chris@49
|
81
|
Chris@49
|
82
|
Chris@49
|
83 //! @}
|