num_entity</span> <span class="o">+</span> <span class="bp">self</span><span class="o">.</span><span class="n">num_relation</span><span class="p">)</span>

        <span class="k">return</span> <span class="n">score</span></div></div>


<div class="viewcode-block" id="RotatE"><a class="viewcode-back" href="../../../api/models.html#torchdrug.models.RotatE">[docs]</a><span class="nd">@R</span><span class="o">.</span><span class="n">register</span><span class="p">(</span><span class="s2">"models.RotatE"</span>