code: def forward(self, text, text_lengths):
      # B, L -> B, L
      emb = self.embed(text)
      emb = self.pos_enc(emb.transpose(1, 2))
      x = self.FFT(emb, text_lengths)
      x = self.out_layer(x).squeeze(-1)
      return x
  
imports: from torch import nn
  from TTS.tts.layers.generic.pos_encoding import PositionalEncoding
  from TTS.tts.layers.generic.transformer import FFTransformerBlock
label_desc: Use image node_id 2 for calling the DurationPredictor obj's underlying member method code with example usage: obj.forward(text, text_lengths) and returns: x