diff --git a/torchnlp/utils.py b/torchnlp/utils.py index 339a9b2..a2446c0 100644 --- a/torchnlp/utils.py +++ b/torchnlp/utils.py @@ -100,7 +100,7 @@ def pad_batch(batch, padding_index=PADDING_INDEX, dim=0): batch (:class:`list` of :class:`torch.Tensor`): Batch of tensors to pad. padding_index (int, optional): Index to pad tensors with. dim (int, optional): Dimension on to which to concatenate the batch of tensors. - + Returns torch.Tensor, list of int: Padded tensors and original lengths of tensors. """