View source
on GitHub
Reads out a tfgnn.Field from the GraphTensor root (i.e. first) node.
runner.RootNodeLabelFn(
node_set_name: tfgnn.NodeSetName,
*,
feature_name: tfgnn.FieldName = tfgnn.HIDDEN_STATE,
**kwargs
)