pub type Batch<B> = HashMap<String, Tensor<B>>;
A single batch: maps names (e.g. “input”, “target”) to tensors.
pub struct Batch<B> { /* private fields */ }