pulsatrix
Loading...
Searching...
No Matches
autograd.hpp File Reference

Reverse-mode autodiff – walks a ComputationGraph backward, accumulating gradients via per-node backward functions supplied by the caller. More...

#include <functional>
#include <unordered_map>
#include <utility>
#include <vector>
#include "pulsatrix/computation_graph.hpp"
#include "pulsatrix/node.hpp"
#include "pulsatrix/tensor.hpp"
Include dependency graph for autograd.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  pulsatrix::Autograd
 Computes gradients by walking a ComputationGraph in reverse topological order. More...
 

Namespaces

namespace  pulsatrix
 

Detailed Description

Reverse-mode autodiff – walks a ComputationGraph backward, accumulating gradients via per-node backward functions supplied by the caller.