File size: 365 Bytes
d758c99 |
1 2 3 4 5 6 7 8 9 |
import torch
def pad_and_tensorize_sequence(sequences, padding_value):
max_size = max([len(sequence) for sequence in sequences])
padded_sequences = []
for sequence in sequences:
padded_sequence = sequence + [padding_value] * (max_size - len(sequence))
padded_sequences.append(padded_sequence)
return torch.tensor(padded_sequences, dtype=torch.long) |