_from_arg</span><span class="p">(</span><span class="n">arg</span><span class="o">=</span><span class="n">radius_init</span><span class="p">,</span>
                                              <span class="n">size</span><span class="o">=</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">n_elements</span><span class="p">,),</span>
                                              <span class="n">default</span><span class="o">=</span><span class="n">torch</span><s