cppdl/tensor.hpp at master · jeanmtr/cppdl · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#include <iostream>
#include <vector>
#include <assert.h>
#include <cmath>
#include <memory>
#include <random>


//TODO: using templates
class Tensor{
   std::shared_ptr<std::vector<double>> data;
  public:
     std::vector<int> shape;
     std::vector<int> stride;
     int dim;
     Tensor(const std::vector<int>& shape);
     Tensor();
     double& get(const std::vector<int>& pos);
     const double& get(const std::vector<int>& pos) const;
     double& get();
     const double& get() const;
     int effectiveDim() const;
     //double get(vector<int> pos); idk
     Tensor operator+(const Tensor& other);
     Tensor operator+(const double other);
     Tensor operator*(const Tensor& other);
     Tensor operator*(const double other);
     Tensor operator-(const Tensor& other);
     Tensor operator-();
     Tensor sum();
     Tensor sum(std::vector<int> axes);
     Tensor mm(const Tensor& other);
     Tensor sigmoid();
     Tensor power(double exp);
     Tensor sigmoidDeriv();
     void printShape();
     Tensor transpose();
     void broadcast(const Tensor& other, int until, Tensor* out1, Tensor* out2) const;
     void fillRandom();
};


template <typename Func>
void iterate(const std::vector<int>& shape, const std::vector<int>& stride, Func&& fn);