9;torch.Tensor&#39;</span><span class="p">],</span>
    <span class="n">Tuple</span><span class="p">[</span><span class="s1">&#39;torch.Tensor&#39;</span><span class="p">,</span> <span class="o">...</span><span class="p">],</span>
<span class="p">],</span>
              <span class="n">query_tokens</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">List</span><span class="p">[</span><span class="s1">&#39;str&#39;</span><span class="p">]],</span> <span class