Skip to content
Snippets Groups Projects
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