_size</span><span class="p">,</span> <span class="n">nnz</span><span class="p">):</span>
        <span class="sd">&quot;&quot;&quot;</span>
<span class="sd">        Approximately solves :math:`\\min_x ||Ax - b||_2 s.t. x_i \in \{0,1\}, ||x||_0 = nnz`</span>
<span class="sd">        </span>
<span class="sd">        Parameters</span>
<span class="sd">        ----------</span>
<span class="sd">        A: torch.Tensor</span>
<span class="sd">            Design matrix of size (d, n)</span>
<span class="sd">     