(l, value=PAD_token)
| 497 | return list(itertools.zip_longest(*l, fillvalue=fillvalue)) |
| 498 | |
| 499 | def binaryMatrix(l, value=PAD_token): |
| 500 | m = [] |
| 501 | for i, seq in enumerate(l): |
| 502 | m.append([]) |
| 503 | for token in seq: |
| 504 | if token == PAD_token: |
| 505 | m[i].append(0) |
| 506 | else: |
| 507 | m[i].append(1) |
| 508 | return m |
| 509 | |
| 510 | # Returns padded input sequence tensor and lengths |
| 511 | def inputVar(l, voc): |