Class: RubyZero::NN::Layers::Embedding

Inherits:

Affine

• Object
• Model
• Layer
• Affine
• RubyZero::NN::Layers::Embedding

Defined in:

lib/rubyzero/nn/layers/embedding.rb

Instance Method Summary

• #initialize(vocab_size, hidden_dim) ⇒ Embedding constructor

  A new instance of Embedding.

Methods inherited from Affine

#forward

Methods inherited from Model

#__get_str__, #call, #eval, #forward, #inspect, load, #parameters, #save, #train

Constructor Details

#initialize(vocab_size, hidden_dim) ⇒ Embedding

Returns a new instance of Embedding.

# File 'lib/rubyzero/nn/layers/embedding.rb', line 3

def initialize(vocab_size, hidden_dim)
    super(vocab_size, hidden_dim, bias: false)
end