Types

Type aliases used across the WarpConvNet API.

Defined in warpconvnet/types.py.

NestedTensor

Alias for torch.Tensor. Used to annotate tensors that represent nested/ragged data (e.g., a batch of variable-length point clouds packed into a single tensor).

from warpconvnet.types import NestedTensor

IterableTensor

IterableTensor = Union[Tensor, List[Tensor], Tuple[Tensor, ...], NestedTensor]

Annotates arguments that accept either a single tensor or a sequence of tensors. Used in APIs that can take a list of per-sample tensors or a pre-concatenated batch tensor.

from warpconvnet.types import IterableTensor