sparse_tensor.h

Go to the documentation of this file.

#ifndef __SPARSE_TENSOR_H__
#define __SPARSE_TENSOR_H__

namespace CTF {
  template<typename dtype=double>
  class Sparse_Tensor {
    public:
      Tensor<dtype> * parent;
      std::vector<int64_t > indices;
      dtype scale;

      Sparse_Tensor();
      
      Sparse_Tensor(std::vector<int64_t >   indices,
                    Tensor<dtype> * parent);

      Sparse_Tensor(int64_t                 n,
                    int64_t       *         indices,
                    Tensor<dtype> * parent);

      void write(dtype   alpha, 
                 dtype * values,
                 dtype   beta); 

      // C++ overload special-cases of above method
      void operator=(std::vector<dtype> values); 
      void operator+=(std::vector<dtype> values); 
      void operator-=(std::vector<dtype> values); 
      void operator=(dtype * values); 
      void operator+=(dtype * values); 
      void operator-=(dtype * values); 

      void read(dtype   alpha, 
                dtype * values,
                dtype   beta); 

      // C++ overload special-cases of above method
      operator std::vector<dtype>();
      operator dtype*();
  };
}

#include "sparse_tensor.cxx"
#endif