gradient.h File Reference

Gradient related operations. More...

#include "concepts/parameter.h"
#include "util/vector_util.h"

Go to the source code of this file.

Namespaces
namespace	bbm

Functions
template<typename PARAM > requires concepts::diff_parameter<PARAM>
void	track_gradients (PARAM &&param, bool toggle=true)
	Enable gradients for a parameter set.
template<typename PARAM > requires concepts::diff_parameter<PARAM>
auto	get_gradients (PARAM &&param)
	Get the gradient from a parameter set.
template<typename PARAM > requires concepts::diff_parameter<PARAM>
auto	detach_gradients (PARAM &&param)
	Get the detached values from a parameter set.
template<typename PARAM > requires concepts::diff_parameter<PARAM>
void	forward_gradients (PARAM &&param)
	Forward computation of gradients on a parameter set.
template<typename T >
void	backward_gradients (T &&t)
	backward computations => passthrough to backbone
template<typename PARAM > requires concepts::diff_parameter<PARAM>
bool	all_gradients_tracked (PARAM &&param)
	Check if gradients are tracked.
template<typename PARAM > requires concepts::diff_parameter<PARAM>
bool	any_gradients_tracked (PARAM &&param)

Detailed Description

Gradient related operations.

Enable gradients,detach, forward, backward, and get gradients types that fullfill concepts::diff_parameter