GradState

class GradState<T : DType, V>(var requiresGrad: Boolean = false, var grad: Tensor<T, V>? = null)(source)

Mutable holder for gradient state associated with a tensor instance.

Constructors

constructor(requiresGrad: Boolean = false, grad: Tensor<T, V>? = null)

Accumulate a gradient tensor onto this state.

Clear the stored gradient.