(sequences, padding_value)
| 141 | create_tensor_by_stacking.attribute_warning = set([]) |
| 142 | |
| 143 | def pad_sequences(sequences, padding_value): |
| 144 | max_size = max([len(sequence) for sequence in sequences]) |
| 145 | padded_sequences = [] |
| 146 | for sequence in sequences: |
| 147 | padded_sequence = sequence + [padding_value] * (max_size - len(sequence)) |
| 148 | padded_sequences.append(padded_sequence) |
| 149 | return padded_sequences |
| 150 | |
| 151 | def get_range_vector(size: int, device) -> torch.Tensor: |
| 152 | """ |