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

Tensor math OpenCL lib

Browse Source
This commit is contained in:
Степанов Платон
2025-11-19 18:05:11 +04:00
parent c1874212ae
commit bd8b26c35a
12 changed files with 361 additions and 355 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/tensor/tensor.hpp
View File

@@ -3,6 +3,7 @@
#include <array>
#include <cstddef>
#include <string>
#include <vector>
template <typename T, int Dim> class Tensor;
@@ -16,6 +17,8 @@ protected:
void checkItHasSameShape(const ITensor &other) const;
void checkAxisInDim(int axis) const;
std::string format(std::vector<T> data) const;
public:
typedef class Tensor<T, Dim> Tensor;
@@ -35,8 +38,8 @@ public:
Tensor &transpose(int axis_a, int axis_b);
Tensor &t();
virtual Tensor operator+() = 0;
virtual Tensor operator-() = 0;
virtual Tensor operator+() const = 0;
virtual Tensor operator-() const = 0;
virtual Tensor &operator+=(const T scalar) = 0;
virtual Tensor &operator*=(const T scalar) = 0;