class TorchText::NN::InProjContainer

Public Class Methods

new(query_proj, key_proj, value_proj) click to toggle source
Calls superclass method
# 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

Public Instance Methods

forward(query, key, value) click to toggle source
# 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