一种表现为单个张量的 dict[str, torch.Tensor] 类型，以及可编译的批量 dataclass 类 TensorClass
A dict[str, torch.Tensor] that behaves like one tensor, and TensorClass: typed, batched dataclasses that compile.
