INNER CODE UNIT · Python

segment_arange

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

def segment_arange(
    ranges: torch.Tensor,
    start: int,
    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 start: int
    :param seq_len:
    :param pad_value:
    :param dtype:
    :param no_cuda_kernel:
    :return:

View source record →