(data, pad_id, width=-1)
| 56 | |
| 57 | |
| 58 | def _pad(data, pad_id, width=-1): |
| 59 | if width == -1: |
| 60 | width = max(len(d) for d in data) |
| 61 | rtn_data = [d + [pad_id] * (width - len(d)) for d in data] |
| 62 | return rtn_data |
| 63 | |
| 64 | |
| 65 | def gen_bert_vector(data, pad_size=200, ): |