Package io.github.kirstenali.deepj.optimisers

Class Parameter

java.lang.Object

io.github.kirstenali.deepj.optimisers.Parameter

public final class Parameter
extends Object

Simple mutable parameter holder for optimizers. Gradients are expected to be accumulated into grad.

Field Summary

Fields

Modifier and Type	Field	Description
Tensor	grad
Tensor	value

Constructor Summary

Constructors

Constructor	Description
Parameter(Tensor value)

Method Summary

Modifier and Type	Method	Description
void	zeroGrad()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

value

public Tensor value

grad

public Tensor grad

Constructor Details

Parameter

public Parameter(Tensor value)

Method Details

zeroGrad

public void zeroGrad()