Class ModuleList

List of modules. Use this if you want to store submodules in a list

Hierarchy

Module
- ModuleList

Index

Constructors

constructor

Properties

Methods

Constructors

constructor

new ModuleList(modules?: Module[]): ModuleList

Inherited from Module.constructor
- Defined in model/basic.ts:114
Parameters
- modules: Module[] = ...
Returns ModuleList

Properties

backend

backend: Backend = 'CPU'

mode

mode: Mode = 'train'

modules

modules: Module[] = ...

Methods

forward

forward(inputs: Tensor<any>[]): Promise<Tensor<any>[]>

Overrides Module.forward
- Defined in model/basic.ts:120
Parameters
- inputs: Tensor<any>[]
Returns Promise<Tensor<any>[]>

get

get(index: number): Module

- Defined in model/basic.ts:128
Parameters
- index: number
Returns Module

getParameters

getParameters(): Variable<any>[]

Inherited from Module.getParameters
- Defined in model/module.ts:34
Returns Variable<any>[]

getSubModules

getSubModules(): Module[]

Overrides Module.getSubModules
- Defined in model/basic.ts:124
Returns Module[]

pop

pop(): undefined | Module

- Defined in model/basic.ts:140
Returns undefined | Module

push

push(module: Module): void

- Defined in model/basic.ts:136
Parameters
- module: Module
Returns void

set

set(index: number, module: Module): void

- Defined in model/basic.ts:132
Parameters
- index: number
- module: Module
Returns void

toBackend

toBackend(backend: Backend): Promise<void>

Inherited from Module.toBackend
- Defined in model/module.ts:54
Parameters
- backend: Backend
Returns Promise<void>

toCPU

toCPU(): Promise<void>

Inherited from Module.toCPU
- Defined in model/module.ts:64
Returns Promise<void>

toGPU

toGPU(): Promise<void>

Inherited from Module.toGPU
- Defined in model/module.ts:98
Returns Promise<void>

toWASM

toWASM(): Promise<void>

Inherited from Module.toWASM
- Defined in model/module.ts:81
Returns Promise<void>

Generated using TypeDoc