Fix small layer issues

This commit is contained in:
2025-11-22 00:33:51 +01:00
parent 4c8b2ef537
commit 104d6ea33d
6 changed files with 12 additions and 4 deletions

View File

@@ -19,4 +19,10 @@ Add::~Add() {}
CUDANet::Tensor&
Add::forward(CUDANet::Tensor& input_a, CUDANet::Tensor& input_b) {
output.zero();
backend->add(
input_a,
input_b,
output
);
return output;
}