Class TorchFunction`Abstract`

Index

Constructors

Properties

_retained_tensors id next_functions opName 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[] = []

opName

opName: string = ''

saved_tensors

saved_tensors: Tensor[] = []

Methods

`Protected` `Abstract`_backward

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

`Protected` `Abstract`_forward

_forward(...args: ArgumentType[]): Tensor
Parameters
- ...args: ArgumentType[]
Returns Tensor
- Defined in functions/base.ts:26

backward

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

forward

forward(...args: ArgumentType[]): Tensor
Parameters
- ...args: ArgumentType[]
Returns Tensor
- Defined in functions/base.ts:29