![]() |
NeuZephyr
Simple DL Framework
|
This namespace contains standard nodes used in computational graphs for neural networks. More...
Classes | |
class | InputNode |
Represents an input node in a computational graph. More... | |
class | OutputNode |
Base class for loss function nodes in a computational graph. More... | |
This namespace contains standard nodes used in computational graphs for neural networks.
The nz::nodes::io
namespace includes the basic building blocks for a computational graph, such as input nodes and output nodes. These nodes serve as the primary interface for data flow in neural networks.
This namespace includes the following classes:
These nodes are designed to be used in a larger computational graph where data flows from one node to another during the forward and backward passes. The input node holds the input data and passes it forward, while the output node computes the loss and manages the gradient flow during backpropagation.
InputNode
does not perform computations, only providing data for the graph.OutputNode
serves as the base class for more specific loss functions (e.g., Mean Squared Error, Cross-Entropy).