Class AttentionWrapper

Hierarchy

AttentionWrapper

Index

Constructors

constructor

Properties

Methods

Constructors

constructor

new AttentionWrapper(cells: tf.LSTMCellFunc[], attnLength: number, attnSize: number): AttentionWrapper

- Defined in src/music_rnn/attention.ts:53
Parameters
- cells: tf.LSTMCellFunc[]
- attnLength: number
- attnSize: number
Returns AttentionWrapper

Properties

attnBias

attnBias: tf.Tensor1D

attnInputBias

attnInputBias: tf.Tensor1D

attnInputMatrix

attnInputMatrix: tf.Tensor2D

attnLength

attnLength: number

attnMatrix

attnMatrix: tf.Tensor2D

attnOutputBias

attnOutputBias: tf.Tensor1D

attnOutputMatrix

attnOutputMatrix: tf.Tensor2D

attnSize

attnSize: number

attnV

attnV: tf.Tensor2D

attnW

attnW: tf.Tensor4D

cells

cells: tf.LSTMCellFunc[]

Methods

call

call(input: tf.Tensor2D, c: tf.Tensor2D[], h: tf.Tensor2D[], state: AttentionState): AttentionWrapperOutput

- Defined in src/music_rnn/attention.ts:80
Parameters
- input: tf.Tensor2D
- c: tf.Tensor2D[]
- h: tf.Tensor2D[]
- state: AttentionState
Returns AttentionWrapperOutput

initState

initState(): AttentionState

- Defined in src/music_rnn/attention.ts:73
Returns AttentionState

initialize

initialize(vars: {}): void

- Defined in src/music_rnn/attention.ts:59
Parameters
- vars: {}
  - [varName: string]: Tensor
Returns void

Static isWrapped

isWrapped(vars: {}): boolean

- Defined in src/music_rnn/attention.ts:51
Parameters
- vars: {}
  - [varName: string]: Tensor
Returns boolean

Generated using TypeDoc