module.hpp File Reference

Abstract base every layer subclasses – NVI forward(), pure-virtual LRP contract. More...

#include <optional>
#include <stdexcept>
#include <string>
#include <utility>
#include <vector>
#include "pulsatrix/assert.hpp"
#include "pulsatrix/autograd.hpp"
#include "pulsatrix/computation_graph.hpp"
#include "pulsatrix/lrp_rule_config.hpp"
#include "pulsatrix/node.hpp"
#include "pulsatrix/op_type.hpp"
#include "pulsatrix/tensor.hpp"

Include dependency graph for module.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	pulsatrix::ParamRef
	A trainable parameter and its accumulated gradient, as owned by some Module. More...
struct	pulsatrix::NamedParamRef
	A parameter together with its hierarchical, dot-separated name relative to the module that reported it (weight, mha.q_proj.bias, 0.weight). More...
class	pulsatrix::Module
	Base class for every layer type (LinearModule, Conv2DModule, activations, ...). More...

Namespaces
namespace	pulsatrix

Functions
void	pulsatrix::append_named_parameters (std::vector< NamedParamRef > &out, const std::string &prefix, Module &child)
	Appends child's named parameters to out, each renamed to prefix.name – the one step every container's named_parameters() repeats per child.

Detailed Description

Abstract base every layer subclasses – NVI forward(), pure-virtual LRP contract.