GradMode (DeepJ 0.13.0-alpha API)

java.lang.Object

io.github.kirstenali.deepj.autograd.GradMode

All Implemented Interfaces:: AutoCloseable

public final class GradMode extends Object implements AutoCloseable

Thread-local, nestable scope for computations that must not build a backward graph.

Method Summary

Modifier and Type

Method

Description

void

close()

static boolean

isEnabled()

static GradMode

noGrad()

Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details
- isEnabled
  
  public static boolean isEnabled()
- noGrad
  
  public static GradMode noGrad()
- close
  
  public void close()
  
  Specified by:
  
  close in interface AutoCloseable