torch_geometric.nn
时间: 2023-04-29 19:03:04 浏览: 150
torch_cluster-1.6.0+pt113cpu-cp310-cp310-linux_x86_64.whl.zip
torch_geometric.nn是一个基于PyTorch的图神经网络库,用于处理图数据。它提供了一系列的图卷积网络、图注意力网络、图池化等模型,可以用于节点分类、图分类、链接预测等任务。同时,它也提供了一些常用的图数据处理工具,如图的读取、转换、可视化等。
阅读全文