">(</span><span class="n">gain_norms</span><span class="p">[</span><span class="n">zero_x</span><span class="p">])</span>
            <span class="n">actual_index</span> <span class="o">=</span> <span class="n">zero_x</span><span class="p">[</span><span class="n">argmin</span><span class="p">]</span><span class="o">.</span><span class="n">item</span><span class="p">()</span>

            <span class="c1"># Add this column to the memoized Ax. Set this weight to 1.</span>
            <span class="n">x</span><