Class: Chainer::Links::Connection::EmbedID

Inherits:
Chainer::Link show all
Defined in:
lib/chainer/links/connection/embed_id.rb

Instance Attribute Summary collapse

Attributes inherited from Chainer::Link

#name

Instance Method Summary collapse

Methods inherited from Chainer::Link

#cleargrads, #del_attr, #init_scope, #namedlinks, #namedparams, #params, #register_persistent, #serialize, #set_attr, #within_init_scope

Constructor Details

#initialize(in_size, out_size, initial_w: nil, ignore_label: nil) ⇒ EmbedID

Returns a new instance of EmbedID.



7
8
9
10
11
12
13
14
15
 
# File 'lib/chainer/links/connection/embed_id.rb', line 7

def initialize(in_size, out_size, initial_w: nil, ignore_label: nil)
  super()
  @ignore_label = ignore_label

  init_scope do
    initial_w ||= Chainer::Initializers::Normal.new(scale: 1.0)
    @w = Chainer::Parameter.new(initializer: initial_w, shape: [in_size, out_size])
  end
end

Instance Attribute Details

#w ⇒ Object (readonly)

Returns the value of attribute w.



5
6
7
 
# File 'lib/chainer/links/connection/embed_id.rb', line 5

def w
  @w
end

Instance Method Details

#call(x) ⇒ Object 



17
18
19
 
# File 'lib/chainer/links/connection/embed_id.rb', line 17

def call(x)
  Chainer::Functions::Connection::EmbedIDFunction.embed_id(x, @w, ignore_label: @ignore_label)
end