Package io.github.kirstenali.deepj.autograd

Class GradientState

java.lang.Object

io.github.kirstenali.deepj.autograd.GradientState

public final class GradientState
extends Object

Runtime-only state; never serialized with a tensor or checkpoint.

Constructor Summary

Constructors

Constructor	Description
GradientState()

Method Summary

Modifier and Type	Method	Description
void	beforeMutation()
void	bindGradient(Tensor value)
Tensor	gradient()
GradientState	newAlias()	Independent autograd metadata sharing only the physical storage's mutation version.
io.github.kirstenali.deepj.autograd.BackwardNode	node()
void	require(boolean value)
boolean	required()
boolean	savedForBackward()
boolean	storageSavedForBackward()
long	version()

Methods inherited from class java.lang.Object

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

Constructor Details

GradientState

public GradientState()

Method Details

required

public boolean required()

version

public long version()

gradient

public Tensor gradient()

node

public io.github.kirstenali.deepj.autograd.BackwardNode node()

newAlias

public GradientState newAlias()

Independent autograd metadata sharing only the physical storage's mutation version.

storageSavedForBackward

public boolean storageSavedForBackward()

require

public void require(boolean value)

bindGradient

public void bindGradient(Tensor value)

savedForBackward

public boolean savedForBackward()

beforeMutation

public void beforeMutation()