diff options
Diffstat (limited to 'src/armadillo/include/armadillo_bits/fn_spones.hpp')
-rw-r--r-- | src/armadillo/include/armadillo_bits/fn_spones.hpp | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/src/armadillo/include/armadillo_bits/fn_spones.hpp b/src/armadillo/include/armadillo_bits/fn_spones.hpp new file mode 100644 index 0000000..ff45b21 --- /dev/null +++ b/src/armadillo/include/armadillo_bits/fn_spones.hpp @@ -0,0 +1,47 @@ +// SPDX-License-Identifier: Apache-2.0 +// +// Copyright 2008-2016 Conrad Sanderson (http://conradsanderson.id.au) +// Copyright 2008-2016 National ICT Australia (NICTA) +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. +// ------------------------------------------------------------------------ + + +//! \addtogroup fn_spones +//! @{ + + + +//! 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 one +template<typename T1> +arma_warn_unused +inline +SpMat<typename T1::elem_type> +spones(const SpBase<typename T1::elem_type, T1>& X) + { + arma_extra_debug_sigprint(); + + typedef typename T1::elem_type eT; + + const unwrap_spmat<T1> U(X.get_ref()); + + SpMat<eT> out(arma_layout_indicator(), U.M); + + arrayops::inplace_set( access::rwp(out.values), eT(1), out.n_nonzero ); + + return out; + } + + + +//! @} |