mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 22:34:22 +00:00
Fix small layer issues
This commit is contained in:
@@ -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;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user