(l, voc)
| 517 | |
| 518 | # Returns padded target sequence tensor, padding mask, and max target length |
| 519 | def outputVar(l, voc): |
| 520 | indexes_batch = [indexesFromSentence(voc, sentence) for sentence in l] |
| 521 | max_target_len = max([len(indexes) for indexes in indexes_batch]) |
| 522 | padList = zeroPadding(indexes_batch) |
| 523 | mask = binaryMatrix(padList) |
| 524 | mask = torch.BoolTensor(mask) |
| 525 | padVar = torch.LongTensor(padList) |
| 526 | return padVar, mask, max_target_len |
| 527 | |
| 528 | # Returns all items for a given batch of pairs |
| 529 | def batch2TrainData(voc, pair_batch): |
no test coverage detected