The input to a Keras Model can be any nest of tensors, including the kind of tuple you sketched.
For deserialization from tf.train.Example, esp. in the presence of more than one GraphTensor, the prefix= argument of tfgnn.parse_example() might come in handy.
Alternatively, if each example has one graph plus some freestanding features, consider packing the freestanding features into the Graph.context, see the Data Preparation and Sampling guide.
The input to a Keras Model can be any nest of tensors, including the kind of tuple you sketched.
For deserialization from
tf.train.Example
, esp. in the presence of more than one GraphTensor, theprefix=
argument oftfgnn.parse_example()
might come in handy.Alternatively, if each example has one graph plus some freestanding features, consider packing the freestanding features into the
Graph.context
, see the Data Preparation and Sampling guide.