NTensor.hh

Go to the documentation of this file.

 
#pragma once
 
#include <vector>
#include <iostream>
 
 
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&);
 
        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 &);