我有一个训练图嵌入的用例，正在寻找一种在 PyTorch 和 TensorFlow 中实现它的方法。限制是该方法应在计算转移概率时结合边缘权重。有人可以建议最好的框架/库吗？作者的 node2vec 实现仅在 spark 中，但是，我当前的 ML 管道对 PyTorch/TF 更友好。