rsvd.hpp

Go to the documentation of this file.

/*
  mptensor - Parallel Library for Tensor Network Methods
 
  Copyright 2016 Satoshi Morita
 
  mptensor is free software: you can redistribute it and/or modify it
  under the terms of the GNU Lesser General Public License as
  published by the Free Software Foundation, either version 3 of the
  License, or (at your option) any later version.
 
  mptensor is distributed in the hope that it will be useful, but
  WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  Lesser General Public License for more details.
 
  You should have received a copy of the GNU Lesser General Public
  License along with mptensor.  If not, see
  <https://www.gnu.org/licenses/>.
*/
 
#ifndef _TENSOR_RSVD_HPP_
#define _TENSOR_RSVD_HPP_
 
#include "tensor.hpp"
 
namespace mptensor {
 
template <template <typename> class Matrix, typename C>
int rsvd(const Tensor<Matrix, C> &a, const Axes &axes_row, const Axes &axes_col,
         Tensor<Matrix, C> &u, std::vector<double> &s, Tensor<Matrix, C> &vt,
         const size_t target_rank, const size_t oversamp);
 
template <template <typename> class Matrix, typename C>
int rsvd(const Tensor<Matrix, C> &a, const Axes &axes_row, const Axes &axes_col,
         Tensor<Matrix, C> &u, std::vector<double> &s, Tensor<Matrix, C> &vt,
         const size_t target_rank);
 
template <template <typename> class Matrix, typename C, typename Func1,
          typename Func2>
int rsvd(Func1 &multiply_row, Func2 &multiply_col, const Shape &shape_row,
         const Shape &shape_col, Tensor<Matrix, C> &u, std::vector<double> &s,
         Tensor<Matrix, C> &vt, const size_t target_rank,
         const size_t oversamp);
 
template <template <typename> class Matrix, typename C, typename Func1,
          typename Func2>
int rsvd(Func1 &multiply_row, Func2 &multiply_col, const Shape &shape_row,
         const Shape &shape_col, Tensor<Matrix, C> &u, std::vector<double> &s,
         Tensor<Matrix, C> &vt, const size_t target_rank);
 
inline void set_seed(unsigned int seed);
 
}  // namespace mptensor
 
#include "rsvd_impl.hpp"
#endif  // _TENSOR_RSVD_HPP_