Skip to content
Snippets Groups Projects
RandomNetwork.hpp 283 B
Newer Older
Franck Dary's avatar
Franck Dary committed
#ifndef RANDOMNETWORK__H
#define RANDOMNETWORK__H

#include "NeuralNetwork.hpp"

class RandomNetworkImpl : public NeuralNetworkImpl
{
  private :

  long outputSize;

  public :

  RandomNetworkImpl(long outputSize);
  torch::Tensor forward(torch::Tensor input) override;
};

#endif