cogdl.models.nn.pyg_gtn
¶
Module Contents¶
Classes¶
-
class
cogdl.models.nn.pyg_gtn.
GTConv
(in_channels, out_channels, num_nodes)[source]¶ Bases:
torch.nn.Module
-
class
cogdl.models.nn.pyg_gtn.
GTLayer
(in_channels, out_channels, num_nodes, first=True)[source]¶ Bases:
torch.nn.Module
-
class
cogdl.models.nn.pyg_gtn.
GTN
(num_edge, num_channels, w_in, w_out, num_class, num_nodes, num_layers)[source]¶ Bases:
cogdl.models.BaseModel