TensorContractor.h

Go to the documentation of this file.

 
#pragma once
 
#ifndef __TENSOR_CONTRACTOR_H_
#define __TENSOR_CONTRACTOR_H_ 1
 
#include <memory>
 
#include "../Utils/Tensor.h"
#include "TensorNode.h"
 
namespace TensorNetworks {
 
class TensorNetwork;
 
class ITensorContractor {
 public:
  using TensorsMap =
      std::unordered_map<Eigen::Index, std::shared_ptr<TensorNode>>;
 
  virtual TensorsMap InitializeTensors(
      const TensorNetwork &network, Types::qubit_t qubit,
      std::vector<Eigen::Index> &keys,
      std::unordered_map<Eigen::Index, Eigen::Index> &keysKeys,
      bool fillKeys = true, bool contract = true) = 0;
 
  virtual double Contract(const TensorNetwork &network,
                          Types::qubit_t qubit) = 0;
 
  virtual size_t GetMaxTensorRank() const = 0;
 
  virtual void SetMultithreading(bool multithreading = true) = 0;
 
  virtual bool GetMultithreading() const = 0;
 
  virtual std::shared_ptr<ITensorContractor> Clone() const = 0;
};
 
}  // namespace TensorNetworks
 
#endif  // __TENSOR_CONTRACTOR_H_