i find the debug，because my input.shape is [6, 574, 574, 100], lengths.shape is[6, 574], so the pack_padded_sequence(input, lengths) can not run, the error is

‘lengths’ argument should be a 1D CPU int64 tensor

what can i specify it?