Refactor layers

This commit is contained in:
2024-03-19 21:35:05 +01:00
parent 8d14b74f66
commit b6c4b7d2ae
12 changed files with 87 additions and 67 deletions

View File

@@ -13,6 +13,10 @@ namespace CUDANet::Layers {
*/
enum ActivationType { SIGMOID, RELU, SOFTMAX, NONE };
/**
* @brief Utility class that performs activation
*
*/
class Activation {
public: