 class="sd">        &gt;&gt;&gt; y = round_to_zero_ste(x)</span>
<span class="sd">        &gt;&gt;&gt; y</span>
<span class="sd">        tensor([ 1., -1.], grad_fn=&lt;RoundToZeroSteFnBackward&gt;)</span>
<span class="sd">        &gt;&gt;&gt; grad = torch.tensor([0.1, -0.1])</span>
<span class="sd">        &gt;&gt;&gt; y.backward(grad)</span>
<span class="sd">        &gt;&gt;&gt; (x.grad == grad).all().item()</span>
<span class="sd">        True</span>
<span class="sd">    &quot;&quot;&quot;</span>
    <spa