Function sequence_padding

utils/data_util.py:424–437

Numpy函数，将序列padding到同一长度

(inputs, length=None, padding=0, is_float=False)

def sequence_padding(inputs, length=None, padding=0, is_float=False):
    """Numpy函数，将序列padding到同一长度
    """
    if length is None:
        length = max([len(x) for x in inputs])

    outputs = np.array([
        np.concatenate([x, [padding] * (length - len(x))])
        if len(x) < length else x[:length] for x in inputs
    ])

    out_tensor = torch.FloatTensor(outputs) if is_float \
        else torch.LongTensor(outputs)
    return torch.tensor(out_tensor)


def batch_gather(data: torch.Tensor, index: torch.Tensor):