scalation
API
Generated with
scalation/scalation/scalation.modeling/scalation.modeling.autograd/GraphExporter/VarNode

VarNode

scalation.modeling.autograd.GraphExporter.VarNode
case class VarNode(id: String, isParam: Boolean, isOutput: Boolean, shape: List[Int], name: Option[String], grad: Option[TensorD])

A variable node in the exported computation graph.

Value parameters

grad

optional stored gradient tensor

id

unique identifier for the variable

isOutput

whether this variable is the graph’s final output

isParam

whether this variable represents a trainable parameter

name

optional user-defined name

shape

tensor shape of the variable

Attributes

Graph
Supertypes
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any
Show all

Members list

Value members

Inherited methods

def productElementNames: Iterator[String]

An iterator over the names of all the elements of this product.

An iterator over the names of all the elements of this product.

Attributes

Inherited from:
Product
def productIterator: Iterator[Any]

An iterator over all the elements of this product.

An iterator over all the elements of this product.

Attributes

Returns

in the default implementation, an Iterator[Any]

Inherited from:
Product
Generated with