full_connected.hpp 549 B

12345678910111213141516171819202122232425
  1. #ifndef FULL_CONNECTED_LAYER_HPP
  2. #define FULL_CONNECTED_LAYER_HPP
  3. #include <random>
  4. #include "layer.hpp"
  5. namespace Layer{
  6. class FullConnectedLayer:public Layer{
  7. public:
  8. Vector b;
  9. Vector w;
  10. Vector nabla_b;
  11. Vector nabla_w;
  12. FullConnectedLayer(size_t n,size_t m);
  13. ~FullConnectedLayer();
  14. void init_standard();
  15. void init(Real mu,Real sigma);
  16. Vector feed_forward(Vector x) override;
  17. void init_nabla() override;
  18. Vector back_propagation(Vector e) override;
  19. void update(Real eta) override;
  20. };
  21. }
  22. #endif