文件名称：PyTorch Geometric：用于PyTorch的几何深度学习扩展库-python
文件大小：692KB
文件格式：ZIP
更新时间：2024-07-08 12:43:31
机器学习
PyTorch Geometric：用于PyTorch的几何深度学习扩展库 文档 | 纸 | 外部资源 | OGB 示例 PyTorch Geometric (PyG) 是 PyTorch 的几何深度学习扩展库。 它由来自各种已发表论文的各种深度学习图和其他不规则结构的方法组成，也称为几何深度学习。 此外，它包括一个易于使用的小批量加载器，用于许多小型和单个巨型图、多 GPU 支持、大量通用基准数据集（基于简单的接口创建自己的）和有用的转换，既可以学习任意图形，也可以学习 3D 网格或点云。 PyTorch Geometric 使实现图神经网络变得轻而易举（参见此处的随附教程）。 例如，这就是实现边缘卷积层所需的全部： import torch from torch.nn import Sequential as Seq, Linear as Lin, ReLU from torch_geometric.nn import MessagePassing class EdgeConv(MessagePassing): def __init__(self, F_in, F_o