1024 my_input_ids my_attention_mask my_token_type_ids my_output true parallel 10 8 1 mean 512 my_input_ids my_attention_mask my_token_type_ids my_output 101 102 parallel 4 8 1