onmt.FeaturesGenerator

Feature decoder generator. Given RNN state, produce categorical distribution over tokens and features.

Implements .

[src]

onmt.FeaturesGenerator(rnnSize, outputSize, features)

Parameters:

  • rnnSize - Input rnn size.
  • outputSize - Output size (number of tokens).
  • features - table of feature sizes.