mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-07 02:04:26 +00:00
Refactor layers
This commit is contained in:
@@ -13,6 +13,10 @@ namespace CUDANet::Layers {
|
||||
*/
|
||||
enum ActivationType { SIGMOID, RELU, SOFTMAX, NONE };
|
||||
|
||||
/**
|
||||
* @brief Utility class that performs activation
|
||||
*
|
||||
*/
|
||||
class Activation {
|
||||
public:
|
||||
|
||||
|
||||
Reference in New Issue
Block a user