class="o">.</span><span class="n">weight</span><span class="p">,</span> <span class="n">s</span><span class="p">))</span>
<span class="n">m</span><span class="o">.</span><span class="n">register_buffer</span><span class="p">(</span><span class="s1">&#39;weight_sparsity&#39;</span><span class="p">,</span> <span class="n">tensor</span><span class="p">(</span><span class="n">s</span><span class="p">))</span>
<span class="n">m</span><span class="o">.</span><span class="n">register_buffer</span><span class="p">(