INNER CODE UNIT · Python

range_fill

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

def range_fill(
    ranges: torch.Tensor,
    values: torch.Tensor,
    seq_len: int,
    pad_value,
    dtype=torch.long,
    no_cuda_kernel: bool = False
):
    """
    Fill a set of values into some ranges of 1D vector.
    Note: The ranges cannot be overlapped, or some unexpected behavior may appear. Current code does not check it.
    :param ranges: (num_ranges, 2)  begin, end
    :param values: (num_ranges,)
    :param seq_len:
    :param pad_value:
    :param dtype:
    :param no_cuda_kernel:
    :return:

View source record →