Class TorchFunction`Abstract`

Index

Constructors

Properties

_retained_tensors id next_functions saved_tensors

Methods

_backward _forward backward forward

Constructors

constructor

new TorchFunction(): TorchFunction
Returns TorchFunction

Properties

_retained_tensors

_retained_tensors: Tensor[] = []

id

id: number = ...

next_functions

next_functions: TorchFunction[] = []

saved_tensors

saved_tensors: Tensor[] = []

Methods

`Protected` `Abstract`_backward

_backward(dz: Tensor): void
Parameters
- dz: Tensor
Returns void
- Defined in functions/base.ts:22

`Protected` `Abstract`_forward

_forward(...args: (number | boolean | number[] | Tensor)[]): Tensor
Parameters
- ...args: (number | boolean | number[] | Tensor)[]
Returns Tensor
- Defined in functions/base.ts:21

backward

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

forward

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