         self.emb_t = nn.Embedding(seq_len, emb_len)</span>
<span class="c1">#         self.t_idx = torch.arange(0, seq_len, dtype=torch.long)</span>
    
    <span class="k">def</span> <span class="nf">forward</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">x</span><span class="p">):</span>
<span class="c1">#         return self.emb(x) + self.emb_t(self.t_idx)</span>
        <span class="n">x</span> <span class="o">=</span> <span class="bp">self</span><spa