Mercurial > hg > segmenter-vamp-plugin
diff armadillo-3.900.4/include/armadillo_bits/fn_sprandu.hpp @ 49:1ec0e2823891
Switch to using subrepo copies of qm-dsp, nnls-chroma, vamp-plugin-sdk; update Armadillo version; assume build without external BLAS/LAPACK
author | Chris Cannam |
---|---|
date | Thu, 13 Jun 2013 10:25:24 +0100 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/armadillo-3.900.4/include/armadillo_bits/fn_sprandu.hpp Thu Jun 13 10:25:24 2013 +0100 @@ -0,0 +1,83 @@ +// Copyright (C) 2012 Conrad Sanderson +// +// This Source Code Form is subject to the terms of the Mozilla Public +// License, v. 2.0. If a copy of the MPL was not distributed with this +// file, You can obtain one at http://mozilla.org/MPL/2.0/. + + +//! \addtogroup fn_sprandu +//! @{ + + + +//! Generate a sparse matrix with a randomly selected subset of the elements +//! set to random values in the [0,1] interval (uniform distribution) +template<typename obj_type> +inline +obj_type +sprandu + ( + const uword n_rows, + const uword n_cols, + const double density, + const typename arma_SpMat_SpCol_SpRow_only<obj_type>::result* junk = 0 + ) + { + arma_extra_debug_sigprint(); + arma_ignore(junk); + + if(is_SpCol<obj_type>::value == true) + { + arma_debug_check( (n_cols != 1), "sprandu(): incompatible size" ); + } + else + if(is_SpRow<obj_type>::value == true) + { + arma_debug_check( (n_rows != 1), "sprandu(): incompatible size" ); + } + + obj_type out; + + out.sprandu(n_rows, n_cols, density); + + return out; + } + + + +inline +sp_mat +sprandu(const uword n_rows, const uword n_cols, const double density) + { + arma_extra_debug_sigprint(); + + sp_mat out; + + out.sprandu(n_rows, n_cols, density); + + return out; + } + + + +//! Generate a sparse matrix with the non-zero values in the same locations as in the given sparse matrix X, +//! with the non-zero values set to random values in the [0,1] interval (uniform distribution) +template<typename T1> +inline +SpMat<typename T1::elem_type> +sprandu(const SpBase<typename T1::elem_type, T1>& X) + { + arma_extra_debug_sigprint(); + + typedef typename T1::elem_type eT; + + SpMat<eT> out( X.get_ref() ); + + eop_aux_randu<eT>::fill( access::rwp(out.values), out.n_nonzero ); + + return out; + } + + + +//! @}