Skip to content
Snippets Groups Projects
Select Git revision
  • a3f93cb0b54f64bcdd8927595f5fd19720a95f4b
  • master default protected
  • johannes
  • partial_parser
  • Aloui_Dary
  • ignore_punct
  • AC
  • classifier
  • fixhelp
  • libmacaon2
  • error_predictor
  • morpho
  • ssrnn
  • tfparsing
  • silvio
  • tagger_options
  • maca_trans_frame_parser
  • alexis
  • new_config
  • tagparse
  • maca_graph_parser
21 results

train_dynet.cpp

Blame
  • OneWordNetwork.hpp 389 B
    #ifndef ONEWORDNETWORK__H
    #define ONEWORDNETWORK__H
    
    #include "NeuralNetwork.hpp"
    
    class OneWordNetworkImpl : public NeuralNetworkImpl
    {
      private :
    
      torch::nn::Embedding wordEmbeddings{nullptr};
      torch::nn::Linear linear{nullptr};
      int focusedIndex;
    
      public :
    
      OneWordNetworkImpl(int nbOutputs, int focusedIndex);
      torch::Tensor forward(torch::Tensor input) override;
    };
    
    #endif