fix pad_sequence, test=asr

pull/1950/head
huangyuxin 2 years ago
parent ea71fddbde
commit 1cdd41bd03

@ -82,7 +82,7 @@ def pad_sequence(sequences: List[paddle.Tensor],
max_size = sequences[0].size() max_size = sequences[0].size()
# (TODO Hui Zhang): slice not supprot `end==start` # (TODO Hui Zhang): slice not supprot `end==start`
# trailing_dims = max_size[1:] # trailing_dims = max_size[1:]
trailing_dims = max_size[1:] if max_size.ndim >= 2 else () trailing_dims = tuple(max_size[1:].numpy().tolist()) if sequences[0].ndim >= 2 else ()
max_len = max([s.shape[0] for s in sequences]) max_len = max([s.shape[0] for s in sequences])
if batch_first: if batch_first:
out_dims = (len(sequences), max_len) + trailing_dims out_dims = (len(sequences), max_len) + trailing_dims

Loading…
Cancel
Save