torch.tx.ExpressionGraph

export interface ExpressionGraph {
  nodes: GraphNode[];
  edges: GraphEdge[];
  /** The root node of the expression (final output) */
  root: number;
}

nodes(GraphNode[])
edges(GraphEdge[])
root(number): – The root node of the expression (final output)

Structured graph representation of an expression