Class: TorchText::NN::InProjContainer

Inherits:

Torch::NN::Module

Object
Torch::NN::Module
TorchText::NN::InProjContainer

show all

Defined in:: lib/torchtext/nn/in_proj_container.rb

Instance Method Summary collapse

#forward(query, key, value) ⇒ Object
#initialize(query_proj, key_proj, value_proj) ⇒ InProjContainer constructor

A new instance of InProjContainer.

Constructor Details

#initialize(query_proj, key_proj, value_proj) ⇒ `InProjContainer`

Returns a new instance of InProjContainer.

# File 'lib/torchtext/nn/in_proj_container.rb', line 4

def initialize(query_proj, key_proj, value_proj)
  super()
  @query_proj = query_proj
  @key_proj = key_proj
  @value_proj = value_proj
end

Instance Method Details

#forward(query, key, value) ⇒ `Object`



11
12
13

# File 'lib/torchtext/nn/in_proj_container.rb', line 11

def forward(query, key, value)
  [@query_proj.call(query), @key_proj.call(key), @value_proj.call(value)]
end