 class="k">def</span><span class="w"> </span><span class="nf">floor_ste</span><span class="p">(</span><span class="n">x</span><span class="p">:</span> <span class="n">Tensor</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">Tensor</span><span class="p">:</span>
<span class="w">    </span><span class="sd">&quot;&quot;&quot;</span>
<span class="sd">    Function that implements :func:`torch.floor` with a straight-through gradient estimator.</span>

<span class="sd">    Notes:</span>
<