NTensor.hh

Go to the documentation of this file.

 
#pragma once
 
#include <vector>
#include <iostream>
 
 
namespace cadabra {
 
    class NTensor {
        public:
            NTensor(const std::vector<size_t>& shape, double val);
 
            NTensor(const std::vector<double>& vals);
 
            NTensor(double);
 
            NTensor(const NTensor&);
 
            static NTensor linspace(double from, double to, size_t steps);
 
            NTensor& operator=(const NTensor&);
 
            NTensor& operator+=(const NTensor&);
 
            NTensor& operator*=(const NTensor&);
 
            NTensor& pow(const NTensor&);
 
            double  at() const;
            
            double  at(const std::vector<size_t>& indices) const;
 
            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(double (*fun)(double));
 
            friend std::ostream& operator<<(std::ostream&, const NTensor&);
 
            std::vector<size_t> shape;
            std::vector<double> values;
    };
 
    std::ostream& operator<<(std::ostream &, const NTensor &);
 
}