Skip to content
Snippets Groups Projects
Select Git revision
  • 9f9a9f9d2d02ff1408f902d4faf54e15ae9c2fe4
  • master default protected
  • loss
  • producer
4 results

MLP.hpp

Blame
  • MLP.hpp 176 B
    #ifndef MLP__H
    #define MLP__H
    
    #include <torch/torch.h>
    
    class MLPImpl : torch::nn::Module
    {
      public :
    
      MLPImpl(const std::string & topology);
    };
    TORCH_MODULE(MLP);
    
    #endif