INNER CODE UNIT · Python

range_fill_pytorch

microsoft/muzic · museformer/museformer/kernels/range_fill/main.py:77

def range_fill_pytorch(ranges, values, seq_len, pad_value, dtype=torch.long):
    out = torch.full((seq_len,), pad_value, dtype=dtype, device=values.device)
    for idx, (begin, end) in enumerate(ranges):
        out[begin: end] = values[idx]
    return out

View source record →