批量处理许多小图

  • Tree-LSTM [论文] [教程] [PyTorch 代码]:句子具有固有的结构,如果仅将其视为序列,则会忽略这些结构。Tree-LSTM 是一个功能强大的模型,它通过使用先验语法结构(如解析树)来学习表示。训练中的挑战在于,仅仅通过将句子填充到最大长度已不再奏效。不同句子的树具有不同的尺寸和拓扑结构。DGL 通过将这些树添加到更大的容器图中,然后使用消息传递来探索最大并行性,从而解决了这个问题。批量处理是实现此功能的关键 API。

DGL 中的 Tree-LSTM

DGL 中的 Tree-LSTM