Method: Transformers::Distilbert::FFN#forward

Defined in:: lib/transformers/models/distilbert/modeling_distilbert.rb

#forward(input) ⇒ `Object`



164
165
166

# File 'lib/transformers/models/distilbert/modeling_distilbert.rb', line 164

def forward(input)
  TorchUtils.apply_chunking_to_forward(method(:ff_chunk), @chunk_size_feed_forward, @seq_len_dim, input)
end