Tensor.h

Go to the documentation of this file.

 
#ifndef KIDS_TENSOR_H
#define KIDS_TENSOR_H
 
#include <memory>
 
#include "kids/Node.h"
#include "kids/Shape.h"
#include "kids/Types.h"
#include "kids/fmt.h"
 
namespace PROJECT_NS {
 
template <typename T>
class Tensor final : public Node {
   public:
    using SizeType = std::size_t;
    using DataType = T;
 
    Tensor(Shape S, const std::string& info = "") : _shape{S}, _doc_info{info} {
        _type = as_enum<T>();
        _size = _shape.size();
        _data = std::shared_ptr<std::vector<T>>(new std::vector<T>(_size, 0));
    }
 
    virtual std::string repr() {
        std::ostringstream os;
        T*                 ptr = _data->data();
        os << as_str<T>();
        os << FMT(0) << _size;
        os << "\n";
        for (int i = 0; i < _size; ++i) os << FMT(8) << ptr[i];
        return os.str();
    }
 
    virtual std::string help(const std::string& name) { return _doc_info; }
 
    inline T* data() { return _data->data(); }
 
    inline std::size_t size() { return _size; }
 
    inline Shape& shape() { return _shape; }
 
   private:
    std::size_t                     _size;
    Shape                           _shape;
    std::string                     _doc_info;
    std::shared_ptr<std::vector<T>> _data;
};
 
};  // namespace PROJECT_NS
 
#endif  // KIDS_TENSOR_H