Function cat

annotator/oneformer/detectron2/layers/wrappers.py:50–57

Efficient version of torch.cat that avoids a copy if there is only a single element in a list

(tensors: List[torch.Tensor], dim: int = 0)

def cat(tensors: List[torch.Tensor], dim: int = 0):
    """
    Efficient version of torch.cat that avoids a copy if there is only a single element in a list
    """
    assert isinstance(tensors, (list, tuple))
    if len(tensors) == 1:
        return tensors[0]
    return torch.cat(tensors, dim)


def empty_input_loss_func_wrapper(loss_func):