lass="p">):</span>
    <span class="sd">"""</span>
<span class="sd">    Message Passing Neural Network proposed in `Neural Message Passing for Quantum Chemistry`_.</span>

<span class="sd">    This implements the enn-s2s variant in the original paper.</span>

<span class="sd">    .. _Neural Message Passing for Quantum Chemistry:</span>
<span class="sd">        https://arxiv.org/pdf/1704.01212.pdf</span>

<span class="sd">    Parameters:</span>
<span class="sd">        input_dim (int): input dimension</span>