NTensor.hh

Go to the documentation of this file.

 
#pragma once
 
#include <vector>
#include <iostream>
#include <complex>
 
 
namespace cadabra {
 
    class NTensor {
        public:
            explicit NTensor(const std::vector<size_t>& shape, std::complex<double> val);
 
            explicit NTensor(const std::vector<std::complex<double>>& vals);
            explicit NTensor(const std::vector<double>& vals);
 
            NTensor(std::initializer_list<std::complex<double>> vals);
            NTensor(std::initializer_list<double> vals);
 
            explicit NTensor(std::complex<double>);
            explicit NTensor(double);
 
            explicit NTensor(const NTensor&);
 
            NTensor(NTensor&&);
 
            static NTensor linspace(std::complex<double> from, std::complex<double> to, size_t steps);
 
            NTensor& operator=(const NTensor&);
 
            NTensor& operator=(NTensor&&) noexcept;
            NTensor& operator=(const NTensor&&) noexcept;
 
            NTensor& operator+=(const NTensor&);
 
            NTensor& operator*=(const NTensor&);
 
            NTensor& operator*=(const std::complex<double>&);
            NTensor& operator*=(double);
 
            NTensor& pow(const NTensor&);
 
            std::complex<double>  at() const;
            
            std::complex<double>  at(const std::vector<size_t>& indices) const;
 
            std::complex<double>& at(const std::vector<size_t>& indices);
 
 
            NTensor broadcast(std::vector<size_t> new_shape, size_t pos) const;
 
 
            static NTensor outer_product(const NTensor& a, const NTensor& b);
 
            NTensor& apply(std::complex<double> (*fun)(const std::complex<double>&));
 
            bool is_real() const;
 
            bool is_scalar() const;
            
            friend std::ostream& operator<<(std::ostream&, const NTensor&);
 
            std::vector<size_t> shape;
            std::vector<std::complex<double>> values;
    };
 
    std::ostream& operator<<(std::ostream &, const NTensor &);
 
}