search

joshcarty / tfgnn-ogb

Examples using TensorFlow GNN with Open Graph Benchmark datasets.
MIT License
7 stars 0 forks source link

Dynamic GraphTensorSpec #7

Open SidneyLann opened 2 years ago

SidneyLann commented 2 years ago

In schema.py, there define one static graph tensor, how to define a dynamic graph tensor when the number of node type(node_set) is dynamic? ie. 2 or 4 node_sets in a graph tensor.

SidneyLann commented 2 years ago 
dataset = tf.data.Dataset.from_generator(
    generator, output_signature=graph_type_spec
)

Different graph tensors will use different graph_type_spec, how to use dynamic graph_type_spec?