Class AccumulateGrad

Hierarchy

UnaryFunction
- AccumulateGrad

Index

Constructors

Properties

_retained_tensors id next_functions saved_tensors variable

Methods

_backward _forward backward forward

Constructors

constructor

new AccumulateGrad(): AccumulateGrad
Returns AccumulateGrad
Inherited from UnaryFunction.constructor

Properties

_retained_tensors

_retained_tensors: Tensor[] = []

id

id: number = ...

next_functions

next_functions: TorchFunction[] = []

saved_tensors

saved_tensors: Tensor[] = []

variable

variable: Tensor

Methods

`Protected`_backward

_backward(dz: number | Tensor): void
Parameters
- dz: number | Tensor
Returns void
Overrides UnaryFunction._backward
- Defined in functions/base.ts:93

`Protected`_forward

_forward(variable: Tensor): Tensor
Parameters
- variable: Tensor
Returns Tensor
Overrides UnaryFunction._forward
- Defined in functions/base.ts:88

backward

backward(dz: Tensor): void
Parameters
- dz: Tensor
Returns void
Inherited from UnaryFunction.backward
- Defined in functions/base.ts:45

forward

forward(...args: (number | boolean | number[] | Tensor)[]): Tensor
Parameters
- ...args: (number | boolean | number[] | Tensor)[]
Returns Tensor
Inherited from UnaryFunction.forward
- Defined in functions/base.ts:24