Complete tensors math

src/math/tensor/tensor.hpp

@@ -0,0 +1,67 @@
+#pragma once
+
+#include <stdexcept>
+#include <vector>
+
+std::random_device rd;
+std::mt19937 gen(rd());
+
+class ITensor {
+protected:
+  std::vector<int> shape;
+
+  void validateDimensions(const std::vector<int> &shape) const {
+    if (shape.empty())
+      throw std::invalid_argument("Tensor shape cannot be empty");
+    for (size_t i = 0; i < shape.size(); ++i) {
+      if (shape[i] <= 0)
+        throw std::invalid_argument(
+            "All tensor dimensions must be positive, but dimension " +
+            std::to_string(i) + " is " + std::to_string(shape[i]));
+    }
+  };
+
+public:
+  ITensor(const std::vector<int> &shape) : shape(shape) {}
+  ITensor(const ITensor &) = default;
+  ITensor &operator=(const ITensor &) = default;
+  ITensor(ITensor &&other) = default;
+  ITensor &operator=(ITensor &&other) = default;
+
+  const std::vector<int> &getShape() const { return shape; }
+  int getDim() const { return static_cast<int>(shape.size()); }
+  size_t getSize() const {
+    size_t size = 1;
+    for (int dim : shape)
+      size *= dim;
+    return size;
+  };
+
+  enum class Type { SCALAR, VECTOR, MATRIX, TENSOR3 };
+  Type getType() const { return static_cast<Type>(shape.size()); };
+};
+
+class ITensor0 {};
+
+class ITensor1 {
+public:
+  virtual int getSize() const = 0;
+};
+
+class ITensor2 {
+public:
+  ITensor2() = default;
+  ITensor2(const ITensor2 &) = default;
+  ITensor2 &operator=(const ITensor2 &) = default;
+  ITensor2(ITensor2 &&other) = default;
+  ITensor2 &operator=(ITensor2 &&other) = default;
+
+  virtual int getRows() const = 0;
+  virtual int getCols() const = 0;
+};
+
+class ITensor3 {};
+
+typedef ITensor0 IScalar;
+typedef ITensor1 IVector;
+typedef ITensor2 IMatrix;