def is_jagged(self) -> bool:
        """Whether the Tensor is jagged (the first dim is JaggedIntVar)."""
        return len(self._attrs["shape"]) > 0 and isinstance(
            self._attrs["shape"][0], JaggedIntVar
        )
Whether the Tensor is jagged (the first dim is JaggedIntVar).
python/aitemplate/compiler/base.py