Tensor.hpp

Go to the documentation of this file.

#ifndef TENSOR_H
#define TENSOR_H

#include <string>
#include <memory>
#include <vector>
#include <sstream>
#include <algorithm>

#ifndef NDEBUG
#include <iostream>

#define DBG_PRINT_2D_TENSOR(s, m) std::cout << s << std::endl; \
    for(unsigned r = 0; r < m.getDims()[0]; ++r){ \
        for(unsigned c = 0; c < m.getDims()[1]; ++c){ \
            std::cout << m(r, c); \
            if(c < m.getDims()[1] - 1) std::cout << ", "; \
        } \
        std::cout << std::endl; \
    }

#define DBG_PRINT_1D_TENSOR(s, v) std::cout << s << std::endl; \
    for(unsigned i = 0; i < v.getDims()[0]; ++i){ \
        std::cout << v(i); \
        if(i < v.getDims()[0] - 1) std::cout << ", "; \
    } \
    std::cout << std::endl
#else
#define DBG_PRINT_2D_TENSOR(_, _)
#define DBG_PRINT_1D_TENSOR(_, _)
#endif

#define MAX_SHOWN_ENTRIES (unsigned)20


template<typename T>
class Tensor {
public:

    Tensor(): size(0) {};

    Tensor(const std::vector<unsigned> dims);

    Tensor(const std::vector<unsigned> dims, const std::vector<T> data);

    Tensor(const std::vector<unsigned> dims, const T& defaultValue);

    Tensor(const Tensor<T>& other);

    inline T& operator()(const int i) const;

    inline T& operator()(const int row, const int col) const;

    inline T& operator()(const int row, const int col, const int offset) const;

    Tensor<T>& operator=(const Tensor<T>& rhs);

    const unsigned getSize() const;

    const std::vector<unsigned> getDims() const;

private:

    unsigned size;

    std::vector<unsigned> dims;

    std::shared_ptr<T[]> data;

};

#include "Tensor.tpp"

#endif // TENSOR_H