runner.RootNodeLabelFn

December 14, 2023 ยท View on GitHub

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
)