cogdl.layers.srgcn_module

Module Contents

Functions

act_attention(attn_type)

act_normalization(norm_type)

act_map(act)

class cogdl.layers.srgcn_module.NodeAttention(in_feat)[source]

Bases: torch.nn.Module

forward(self, x, edge_index, edge_attr)[source]
class cogdl.layers.srgcn_module.EdgeAttention(in_feat)[source]

Bases: torch.nn.Module

forward(self, x, edge_index, edge_attr)[source]
class cogdl.layers.srgcn_module.Identity(in_feat)[source]

Bases: torch.nn.Module

forward(self, x, edge_index, edge_attr)[source]
class cogdl.layers.srgcn_module.PPR(in_feat)[source]

Bases: torch.nn.Module

forward(self, x, edge_index, edge_attr)[source]
class cogdl.layers.srgcn_module.HeatKernel(in_feat)[source]

Bases: torch.nn.Module

forward(self, x, edge_index, edge_attr)[source]
cogdl.layers.srgcn_module.act_attention(attn_type)[source]
class cogdl.layers.srgcn_module.NormIdentity[source]

Bases: torch.nn.Module

forward(self, edge_index, edge_attr, N)[source]
class cogdl.layers.srgcn_module.RowUniform[source]

Bases: torch.nn.Module

forward(self, edge_index, edge_attr, N)[source]
class cogdl.layers.srgcn_module.RowSoftmax[source]

Bases: torch.nn.Module

forward(self, edge_index, edge_attr, N)[source]
class cogdl.layers.srgcn_module.ColumnUniform[source]

Bases: torch.nn.Module

forward(self, edge_index, edge_attr, N)[source]
class cogdl.layers.srgcn_module.SymmetryNorm[source]

Bases: torch.nn.Module

forward(self, edge_index, edge_attr, N)[source]
cogdl.layers.srgcn_module.act_normalization(norm_type)[source]
cogdl.layers.srgcn_module.act_map(act)[source]