v class="input_area">
<div class=" highlight hl-ipython3"><pre><span></span><span class="n">drop</span><span class="p">(</span><span class="n">yhat</span><span class="p">)</span>
</pre></div>

    </div>
</div>
</div>

<div class="output_wrapper">
<div class="output">

<div class="output_area">



<div class="output_text output_subarea output_execute_result">
<pre>tensor([[[ 0.4389,  2.0551,  0.6879,  1.2779],
         [-0.5141,  0.7211, -0.9958, -0.6397],
         [-0.3407, -0.3860, -1.0828,  0.0695]],

  