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_output

101

102

parallel

4

8

1

1024

32

512

101

102

103

my_input_ids

my_attention_mask

my_token_type_ids

my_output

parallel

10

8

1