Module computational_graph

Module computational_graph 

Source
Expand description

Computational graph implementation

Structs§

AddNode
Implementation of additive nodes
BranchNode
Branch node implementation
BroadcastNode
Broadcast node implementation
MulNode
Multiplication node implementation
ReciprocalNode
Implementation of reciprocal nodes
SqrtNode
Square root node implementation
SquareNode
Squared node implementation
SubNode
Implementation of negative additive nodes
SumNode
Sum node implementation

Traits§

GraphNode
Trait that defines a computational graph for calculating forward and back propagation of a neural network