pylibwholegraph.torch.create_gnn_layers#

pylibwholegraph.torch.create_gnn_layers(
in_feat_dim,
hidden_feat_dim,
class_count,
num_layer,
num_head,
model_type,
)#