11 lines
190 BLFS
Plaintext
11 lines
190 BLFS
Plaintext
name: jax-scan
|
|
description: Run scan-based RNN forward
|
|
inputs:
|
|
seq: jax.Array
|
|
Wx: jax.Array
|
|
Wh: jax.Array
|
|
b: jax.Array
|
|
outputs:
|
|
hseq: jax.Array
|
|
implementation: jax_skills.rnn_scan
|