Implement concat layer

This commit is contained in:
2024-03-17 21:38:29 +01:00
parent 9aab7cdb10
commit aac0c3a826
9 changed files with 121 additions and 33 deletions

View File

@@ -64,18 +64,6 @@ class ILayer {
* @brief Copy the weights and biases to the device
*/
virtual void toCuda() = 0;
int inputSize;
int outputSize;
float* d_output;
float* d_weights;
float* d_biases;
std::vector<float> weights;
std::vector<float> biases;
};
} // namespace CUDANet::Layers