">scaling_stats_input_concat_dim</span><span class="p">:</span> <span class="nb">int</span><span class="p">,</span>
            <span class="n">tracked_parameter_list</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">torch</span><span class="o">.</span><span class="n">nn</span><span class="o">.</span><span class="n">Parameter</span><span class="p">],</span>
            <span class="n">scaling_shape</span><span class="p">:</span> <span class="n">Tuple</span><sp