base
This is summary documentation for hippynn.graphs.nodes.base module. Click here for full documentation.
API Summary for sub-modules:
Links to Full Documentation:
Core classes and functions for nodes.
This subpackage is separated into separate files to reduce the complexity of understanding the definitions.
For a user, all relevant functions and names should be exposed here.
The basic functionality of a node in a directed graph is defined in node_functions.py as _NodeMethods.
To augment this, the registration of dunder methods is implemented in algebra.py.
Combining these, base.py defines many of the core node classes.
the multi.py module goes on to define a particular core class of import, the MultiNode, which itself represents a single pytorch computation which returns multiple tensors.
Last but not least is parent_expander, which contains optional features which help to simplify the creation nodes using flexible parent types.
Modules
Machinery for nodes to support native python operators such as addition, multiplication, subtraction. |
|
Base nodes for sublcassing. |
|
A base node that provides several output tensors. |
|
Here we define some of the core operations for Nodes in the directed acyclic graph associated with a computation. |
|
Tools for organizing node definitions. |
|