mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-23 14:54:28 +00:00
Migrate Activation layer to Tensor
This commit is contained in:
@@ -30,7 +30,7 @@ class Activation {
|
||||
* @param activation Type of activation
|
||||
* @param length Length of the input
|
||||
*/
|
||||
Activation(ActivationType activation, const int length);
|
||||
Activation(CUDANet::Backend::IBackend* backend, ActivationType activation, const int length);
|
||||
|
||||
/**
|
||||
* @brief Destroy the Activation object
|
||||
|
||||
Reference in New Issue
Block a user