Utility and Meta Modules#
A module that allows adding comments to |
|
A simple bias module that adds a (trainable by default) bias array (default) or scalar to the input. |
|
Module that flattens the input to 1D. |
|
Module that reshapes the input array to a specified shape. |
|
Meta-module that applies a given module to a subset of the input data and optionally passes the remaining input data through unchanged. |
|
Meta-module that applies multiple modules in sequence. |
|
Module that implements a general element-wise function, optionally with trainable parameters and state. |