dlprim::TensorAndGradient Struct Reference

Pair of tensor and its gradient for backpropogation. More...

#include <include/dlprim/tensor.hpp>

Public Attributes
bool	requires_gradient =true
float	accumulate_gradient =0.0
Tensor	data
Tensor	diff
	value

Detailed Description

Pair of tensor and its gradient for backpropogation.

Member Data Documentation

float dlprim::TensorAndGradient::accumulate_gradient =0.0

• accumulate factor. 0 - gradient is overrwritten, 1.0 is fully accumulated

bool dlprim::TensorAndGradient::requires_gradient =true

• set to false to prevent computations of the gradient

---

The documentation for this struct was generated from the following file:

• include/dlprim/tensor.hpp