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

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

@@ -41,18 +41,24 @@ template <ITensor0Type T> class ITensor0Math {};
template <ITensor1Type T> class ITensor1Math {};
template <ITensor2Type T> class ITensor2Math {
template <ITensor2Type M, ITensor1Type V> class ITensor2Math {
public:
virtual T mult(const T &a, const T &b, bool transpose, float bias,
virtual M mult(const M &a, const M &b, bool transpose, const V *bias,
Activation type, float alpha) = 0;
void validateMultDimensions(const T &a, const T &b, bool transpose) const {
void validateMultDimensions(const M &a, const M &b, bool transpose) const {
if ((!transpose && a.getCols() != b.getRows()) ||
(transpose && a.getCols() != b.getCols())) {
throw std::invalid_argument(
"Invalid matrix dimensions for multiplication");
}
};
void validateBiasDimensions(const M &a, const V &b, bool transpose) const {
if ((!transpose && a.getCols() != b.getSize()) ||
(transpose && a.getRows() != b.getSize())) {
throw std::invalid_argument("Invalid matrix bias");
}
};
};
template <ITensor3Type T> class ITensor3Math {};