nip.utils.torch.CatGraphPairDim#
- class nip.utils.torch.CatGraphPairDim(cat_dim: int, pair_dim: int = 0)[source]#
Concatenate the two node sets for each graph pair.
- Parameters:
Methods Summary
__init__
(cat_dim[, pair_dim])Initialize internal Module state, shared by both nn.Module and ScriptModule.
forward
(x)Concatenate the two node sets for each graph pair.
Attributes
T_destination
call_super_init
dump_patches
training
Methods