[Python]填充mask

2019-04-01  本文已影响0人  VanJordan
encoded_inputs = [encoders["inputs"].encode(x) + [1] for x in input_txt]
# pad each input so is they are the same length
biggest_seq = len(max(encoded_inputs, key=len))
for i, text_input in enumerate(encoded_inputs):
    encoded_inputs[i] = text_input + [0 for x in range(biggest_seq - len(text_input))]
上一篇 下一篇

猜你喜欢

热点阅读