class="c1"># ,dtype=torch.float64)</span>

        <span class="c1"># Calculate b * n / val_set_size</span>
        <span class="n">b_k_val</span> <span class="o">=</span> <span class="p">(</span><span class="n">n</span> <span class="o">/</span> <span class="n">val_set_size</span><span class="p">)</span> <span class="o">*</span> <span class="n">b</span>

        <span class="c1"># Stores memoized version of Ax, where x is the vector containing an element from {0,1}</span>
        <span class="n">memoized_Ax