h_model.PyTorchModel.get_num_output_nodes">[docs]</a>
    <span class="k">def</span><span class="w"> </span><span class="nf">get_num_output_nodes</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">inp_size</span><span class="p">):</span>
        <span class="n">temp_input</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">rand</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span 