1.相关包导入
import os
import re
import inspect
import os.path as osp
from uuid import uuid1
from itertools import chain
from inspect import Parameter
from typing import List, Optional, Set
from torch_geometric.typing import Adj, Size
import torch
from torch import Tensor
from jinja2 import Template
from torch_sparse import SparseTensor
from torch_scatter import gather_csr, scatter, segment_csr
from .utils.helpers import expand_left
from .utils.jit import class_from_module_repr
from .utils.typing import (sanitize, split_types_repr, parse_types,
resolve_types)
from .utils.inspector import Inspector, func_header_repr, func_body_repr
2.
2.1 数学表达
x i ( k ) = γ ( k ) ( x i ( k − 1 ) , □ j ∈ N ( i ) ϕ ( k ) ( x i ( k − 1 ) , x j ( k − 1 ) , e j , i ) ) , \mathbf{x}_i^{(k)} = \gamma^{(k)} \left( \mathbf{x}_i^{(k-1)}, \square_{j \in \mathcal{N}(i)} \, \phi^{(k)}\left(\mathbf{x}_i^{(k-1)}, \mathbf{x}_j^{(k-1)},\mathbf{e}_{j,i}\right) \right), xi(k)=γ(k)(xi(k−1),□j∈N(i)ϕ(k)(xi(k−1),xj(k−1),ej,i)),
□ \square □ d表示可微的置换不变函数, e.g., sum, mean or max
γ \gamma γ and ϕ \phi ϕ 表示可微函数,例如 MLP(多层感知器).
2.2三个用户自定义函数
上述三个函数是需要需用户自定义的,即;
1. ϕ \phi ϕ-message()
2. γ \gamma γ -update(),
3. □ − \square- □− 聚合方式, aggr=“add”,aggr="mean"或aggr=“max”
3 源码
3.1 MessagePassing(aggr=“add”, flow=“source_to_target”, node_dim=-2)
aggr (string, optional): 聚合方案 ( "add","mean"或"max")
(default: :`"add"`)
flow (string, optional): 消息传递的流向 ("source_to_target"或"target_to_source") 也就是A→B还是B→A