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_