StepanovPlaton/NeuralNetwork
1
0
Fork 0
mirror of https://github.com/StepanovPlaton/NeuralNetwork.git synced 2026-04-04 04:40:40 +04:00
Code Issues Packages Projects Releases Wiki Activity

Forward with new tensors math

Browse Source
This commit is contained in:
Степанов Платон
2025-11-01 14:22:09 +04:00
parent f1dfe1b335
commit c548c3089b
8 changed files with 192 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/math/tensor/tensor.hpp
View File

@@ -23,10 +23,16 @@ protected:
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;
ITensor(const ITensor &other) : shape(other.shape) {};
ITensor &operator=(const ITensor &other) {
shape = other.shape;
return *this;
};
ITensor(ITensor &&other) : shape(other.shape) {};
ITensor &operator=(ITensor &&other) {
shape = other.shape;
return *this;
};
const std::vector<int> &getShape() const { return shape; }
int getDim() const { return static_cast<int>(shape.size()); }
@@ -50,12 +56,6 @@ public:
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;
};