(l, voc)
| 509 | |
| 510 | # Returns padded input sequence tensor and lengths |
| 511 | def inputVar(l, voc): |
| 512 | indexes_batch = [indexesFromSentence(voc, sentence) for sentence in l] |
| 513 | lengths = torch.tensor([len(indexes) for indexes in indexes_batch]) |
| 514 | padList = zeroPadding(indexes_batch) |
| 515 | padVar = torch.LongTensor(padList) |
| 516 | return padVar, lengths |
| 517 | |
| 518 | # Returns padded target sequence tensor, padding mask, and max target length |
| 519 | def outputVar(l, voc): |
no test coverage detected