(lod)
| 24 | |
| 25 | |
| 26 | def convert_to_offset(lod): |
| 27 | offset = [[0] for i in lod] |
| 28 | for i, level in enumerate(lod): |
| 29 | for seq_len in level: |
| 30 | offset[i].append(offset[i][-1] + seq_len) |
| 31 | return offset |
| 32 | |
| 33 | |
| 34 | def compute_seqpool_sum(x, offset, out, pad_value=0.0): |