Source code for torchsparse.operators

from typing import List

import torch

from torchsparse.tensor import SparseTensor

__all__ = ['cat']


def cat(inputs: List[SparseTensor]) -> SparseTensor:
    """Concatenate a list of sparse tensors.
    
    Args:
        inputs (List[SparseTensor]): A list of sparse tensors.
    
    Returns:
        SparseTensor: The concatenated sparse tensor.
    """
    feats = torch.cat([input.feats for input in inputs], dim=1)
    output = SparseTensor(coords=inputs[0].coords,
                          feats=feats,
                          stride=inputs[0].stride)
    output.cmaps = inputs[0].cmaps
    output.kmaps = inputs[0].kmaps
    return output