std1.0.0-beta-7 SafeC docs

std :: ml::tensor

Struct std::Autograd

#include <std/ml/tensor.h> source

struct Autograd {
    int gradEnabled;
}

Fields

§int gradEnabledsource

1: ops record their backward graph

Methods

§int enabled() constsource · definition
§void set_enabled(int enabled)source · definition
§struct GradGuard no_grad()source · definition

Disables recording and returns a guard that restores the previous mode.

§static &Autograd get()source · definition

The program's Autograd::get state (one for all threads).