Class GradientContext

Hierarchy

GradientContext

Indexable

[key: string]: any

Index

Constructors

Properties

inputsWithGradient needsInputGradient savedTensors

Methods

saveForBackward

Constructors

constructor

new GradientContext(inputs): GradientContext
Parameters
- inputs: FunctionInput[]
Returns GradientContext
- Defined in autograd.ts:16

Properties

inputsWithGradient

inputsWithGradient: (null | Tensor)[]

needsInputGradient

needsInputGradient: boolean[]

savedTensors

savedTensors: Tensor[] = []

Methods

saveForBackward

saveForBackward(...tensors): void
Parameters
- Rest ...tensors: Tensor[]
Returns void
- Defined in autograd.ts:24

Generated using TypeDoc