NDArray.h

Go to the documentation of this file.

 
#pragma once
 
#include <functional>
#include <string>
#include <tuple>
#include <unordered_set>
#include <vector>
 
class NDArray {
private:
  NDArray(std::vector<int> shape, std::vector<int> strides, float *data,
          bool ownsData, std::string label = "", std::string op = "",
          std::vector<std::reference_wrapper<NDArray>> prev = {});
 
  void build_topo(std::unordered_set<NDArray *> &visited, NDArray *arr,
                  std::vector<std::reference_wrapper<NDArray>> &topo);
 
public:
  float *data; 
  std::vector<int> shape; 
  std::vector<int> strides; 
  int ndim; 
  int size = 0; 
  bool ownsData; 
  float *grad; 
  std::string op; 
  std::string label; 
  std::vector<std::reference_wrapper<NDArray>> prev; 
  enum class PrintType { Data, Grad }; 
  std::function<void()> _backward; 
  NDArray(std::vector<int> shape, std::string label = "", std::string op = "",
          std::vector<std::reference_wrapper<NDArray>> prev = {});
 
  void metadata(bool shapeInfo = true, bool stridesInfo = false,
                bool ndimInfo = false, bool sizeInfo = false,
                bool ownsDataInfo = false);
 
  float get(std::vector<int> indices, PrintType type = PrintType::Data) const;
 
  float get(int index, PrintType type = PrintType::Data) const;
 
  void set(std::vector<int> indices, float value);
 
  void set(int index, float value);
 
  NDArray slice(std::vector<std::tuple<int, int>> indices);
 
  bool isContiguous() const;
 
  void reshape(std::vector<int> newShape);
 
  void print(PrintType type = PrintType::Data);
 
  void fillSequential();
 
  void fill(float value);
 
  void zeros();
 
  void ones();
 
  void randint(int low, int high);
 
  void rand();
 
  void rand(float low, float high);
 
  NDArray operator+(NDArray &other);
 
  NDArray operator+(float value);
 
  NDArray operator-(NDArray &other);
 
  NDArray operator-(float value);
 
  NDArray operator*(NDArray &other);
 
  NDArray operator*(float value);
 
  NDArray operator/(NDArray &other);
 
  NDArray operator/(float value);
 
  NDArray operator^(float value);
 
  NDArray element_wise_multiply(NDArray &other);
 
  NDArray element_wise_multiply(float value);
 
  NDArray sum();
 
  NDArray sum(int axis);
 
  void backward();
 
  NDArray clone();
};