Class ModuleDict

Dictionary of modules. Use this if you want to store submodules in a dictionary

Hierarchy

Module
- ModuleDict

Index

Constructors

constructor

Properties

Methods

Constructors

constructor

new ModuleDict(modules?: {}): ModuleDict

Inherited from Module.constructor
- Defined in model/basic.ts:84
Parameters
- modules: {} = ...
  - [name: string]: Module
Returns ModuleDict

Properties

backend

backend: Backend = 'CPU'

mode

mode: Mode = 'train'

modules

modules: {} = ...

Type declaration

[name: string]: Module

Methods

forward

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

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

get

get(key: string): Module

- Defined in model/basic.ts:102
Parameters
- key: string
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:94
Returns Module[]

set

set(key: string, module: Module): void

- Defined in model/basic.ts:106
Parameters
- key: string
- 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