@@ -164,7 +164,7 @@ import Tensor from '../../../util/tensor.js'
164
164
* { type: 'global_max_pool', channel_dim?: number } |
165
165
* { type: 'greater' } |
166
166
* { type: 'greater_or_equal' } |
167
- * { type: 'gru', size: number, return_sequences?: boolean, w_z?: number[][] | Matrix, w_r?: number[][] | Matrix, w_h?: number[][] | Matrix, u_z?: number[][] | Matrix, u_r?: number[][] | Matrix, u_h?: number[][] | Matrix, b_z?: number[][] | Matrix, b_r?: number[][] | Matrix, b_h?: number[][] | Matrix } |
167
+ * { type: 'gru', size: number, return_sequences?: boolean, w_z?: number[][] | Matrix | string , w_r?: number[][] | Matrix | string , w_h?: number[][] | Matrix | string , u_z?: number[][] | Matrix | string , u_r?: number[][] | Matrix | string , u_h?: number[][] | Matrix | string , b_z?: number[][] | Matrix | string , b_r?: number[][] | Matrix | string , b_h?: number[][] | Matrix | string, sequence_dim?: number } |
168
168
* { type: 'hard_elish' } |
169
169
* { type: 'hard_shrink', l?: number } |
170
170
* { type: 'hard_sigmoid', alpha?: number, beta?: number } |
@@ -190,7 +190,7 @@ import Tensor from '../../../util/tensor.js'
190
190
* { type: 'logsigmoid' } |
191
191
* { type: 'lp_pool', p?: number, kernel: number | number[], stride?: number | number[], padding?: number | number[], channel_dim?: number } |
192
192
* { type: 'lrn', alpha?: number, beta?: number, k?: number, n: number, channel_dim?: number } |
193
- * { type: 'lstm', size: number, return_sequences?: boolean, w_z?: number[][] | Matrix, w_in?: number[][] | Matrix, w_for?: number[][] | Matrix, w_out?: number[][] | Matrix, r_z?: number[][] | Matrix, r_in?: number[][] | Matrix, r_for?: number[][] | Matrix, r_out?: number[][] | Matrix, p_in?: number[][] | Matrix, p_for?: number[][] | Matrix, p_out?: number[][] | Matrix, b_z?: number[][] | Matrix, b_in?: number[][] | Matrix, b_for?: number[][] | Matrix, b_out?: number[][] | Matrix } |
193
+ * { type: 'lstm', size: number, return_sequences?: boolean, w_z?: number[][] | Matrix | string , w_in?: number[][] | Matrix | string , w_for?: number[][] | Matrix | string , w_out?: number[][] | Matrix | string , r_z?: number[][] | Matrix | string , r_in?: number[][] | Matrix | string , r_for?: number[][] | Matrix | string , r_out?: number[][] | Matrix | string , p_in?: number[][] | Matrix | string , p_for?: number[][] | Matrix | string , p_out?: number[][] | Matrix | string , b_z?: number[][] | Matrix | string , b_in?: number[][] | Matrix | string , b_for?: number[][] | Matrix | string , b_out?: number[][] | Matrix | string, sequence_dim?: number } |
194
194
* { type: 'matmul' } |
195
195
* { type: 'max' } |
196
196
* { type: 'max_pool', kernel: number | number[], stride?: number | number[], padding?: number | number[], channel_dim?: number } |
@@ -230,7 +230,7 @@ import Tensor from '../../../util/tensor.js'
230
230
* { type: 'reu' } |
231
231
* { type: 'reverse', axis?: number } |
232
232
* { type: 'right_bitshift' } |
233
- * { type: 'rnn', size: number, out_size?: number, activation?: string | object, recurrent_activation?: string | object, return_sequences?: boolean, w_xh ?: number[][] | Matrix, w_hh?: number[][] | Matrix, w_hy ?: number[][] | Matrix, b_xh ?: number[][] | Matrix, b_hh ?: number[][] | Matrix, b_hy ?: number[][] | Matrix } |
233
+ * { type: 'rnn', size: number, activation?: string | object, return_sequences?: boolean, w_x ?: number[][] | Matrix | string, w_h ?: number[][] | Matrix | string, b_x ?: number[][] | Matrix | string, b_h ?: number[][] | Matrix | string, sequence_dim ?: number } |
234
234
* { type: 'rootsig' } |
235
235
* { type: 'round' } |
236
236
* { type: 'rrelu', l?: number, u?: number } |
0 commit comments