nd2py.utils.nn package# Submodules# nd2py.utils.nn.gnn module# class nd2py.utils.nn.gnn.GNNLayer(*args: Any, **kwargs: Any)[source]# Bases: Module __init__(node_dim, edge_dim, out_dim, residual=False)[source]# forward(v, e, edge_list, num_nodes)[source]# v: [N, V, node_dim] (N: SampleNum, V: TotalNodes) e: [N, E, edge_dim] (N: SampleNum, E: TotalEdges) edge_list: [2, E] (第0行: source, 第1行: target) num_nodes: [N] (每个样本的节点数量) return: [N, V, out_dim], [N, E, out_dim] class nd2py.utils.nn.gnn.GNN(*args: Any, **kwargs: Any)[source]# Bases: Module __init__(d_emb, n_layers, node_dim, edge_dim, dropout)[source]# fwd(v, e, edge_list, num_nodes)[source]# forward(v, e, edge_list, num_nodes, chunk_size=None)[source]# nd2py.utils.nn.positional_encoding module# class nd2py.utils.nn.positional_encoding.PositionalEncoding(*args: Any, **kwargs: Any)[source]# Bases: Module __init__(d_emb, dropout=0.2, max_len=5000)[source]# forward(x)[source]#