No, you need tensors of shape 1, num_choices, seq_length
, which is what the code sample will give you with the second line that does the expand_dims.
1 Like
No, you need tensors of shape 1, num_choices, seq_length
, which is what the code sample will give you with the second line that does the expand_dims.