annotate armadillo-3.900.4/include/armadillo_bits/fn_sprandu.hpp @ 84:55a047986812 tip

Update library URI so as not to be document-local
author Chris Cannam
date Wed, 22 Apr 2020 14:21:57 +0100
parents 1ec0e2823891
children
rev   line source
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 //! @}