(batch, output, i)
| 376 | return tf.less(i, tf.shape(batch)[1]) |
| 377 | |
| 378 | def body(batch, output, i): |
| 379 | self_attention_tmp = din_fcn_attention(batch[:, i, :], batch[:, 0:i+1, :], |
| 380 | ATTENTION_SIZE, mask[:, 0:i+1], softmax_stag=1, stag=stag, |
| 381 | mode='LIST') |
| 382 | self_attention_tmp = tf.reduce_sum(self_attention_tmp, 1) |
| 383 | output = output.write(i, self_attention_tmp) |
| 384 | return batch, output, i + 1 |
| 385 | |
| 386 | output_ta = tf.TensorArray(dtype=tf.float32, |
| 387 | size=0, |
nothing calls this directly
no test coverage detected