tical-align: top;"><ul>
        <li><a href="api/models.html#torchdrug.models.KBGAT">KBGAT (in module torchdrug.models)</a>
</li>
        <li><a href="api/data.html#torchdrug.data.key_split">key_split() (in module torchdrug.data)</a>
</li>
        <li><a href="api/layers.html#torchdrug.layers.geometry.KNNEdge">KNNEdge (class in torchdrug.layers.geometry)</a>
</li>
    </ul></td>
    <td style="width: 33%; vertical-align: top;"><ul>
        <li><a href="api/models.html#torchdrug.models.KnowledgeBaseGraphAtte