/span>
</span><span id="L-28"><a href="#L-28"><span class="linenos">28</span></a>        <span class="sd">&quot;&quot;&quot;Initializes the module.</span>
</span><span id="L-29"><a href="#L-29"><span class="linenos">29</span></a><span class="sd">        </span>
</span><span id="L-30"><a href="#L-30"><span class="linenos">30</span></a><span class="sd">        Args:</span>
</span><span id="L-31"><a href="#L-31"><span class="linenos">31</span></a><span class="sd">            input_size: The expected input_size