Package io.github.kirstenali.deepj.autograd

Class Autograd

java.lang.Object

io.github.kirstenali.deepj.autograd.Autograd

public final class Autograd
extends Object

Method Summary

Modifier and Type	Method	Description
static void	backward(Tensor output)
static void	backward(Tensor output, Tensor gradient)
static Tensor	record(Tensor output, BackwardRule rule, Tensor... inputs)
static Tensor	recordSaved(Tensor output, BackwardRule rule, Tensor[] inputs, Tensor... saved)
static boolean	tracks(Tensor... inputs)

Methods inherited from class java.lang.Object

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

Method Details

tracks

public static boolean tracks(Tensor... inputs)

record

public static Tensor record(Tensor output,
 BackwardRule rule,
 Tensor... inputs)

recordSaved

public static Tensor recordSaved(Tensor output,
 BackwardRule rule,
 Tensor[] inputs,
 Tensor... saved)

backward

public static void backward(Tensor output)

backward

public static void backward(Tensor output,
 Tensor gradient)