-
Franck Dary authoredFranck Dary authored
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
#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