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