Sort the batch in the descending order sorted_batch = sorted(batch, key=lambda x: x[0].shape[0], reverse=True) # Get each sequence and pad ...
確定! 回上一頁