FuncNode
scalation.modeling.autograd.GraphExporter.FuncNode
case class FuncNode(id: String, op: String, attrs: Map[String, String], shape: List[Int], depth: Int)
A function node (operation) in the computation graph.
Value parameters
- attrs
-
operator-specific attributes
- depth
-
depth level for layered graph layout
- id
-
unique identifier for the function
- op
-
operation name (e.g., "add", "matmul")
- shape
-
tensor shape of the function output
Attributes
- Graph
-
- Supertypes
-
trait Serializabletrait Producttrait Equalsclass Objecttrait Matchableclass Any
Members list
In this article