abstract_dataloader.ext.types
¶
Data type system framework following the ADL recommendations.
Programming Model
Warning
This module requires optree
to be installed.
abstract_dataloader.ext.types.ArrayLike
¶
Bases: Protocol
Array type, e.g., torch.Tensor | jax.Array | np.ndarray
.
Use this type to specify arbitrary array types.
Source code in src/abstract_dataloader/ext/types.py
abstract_dataloader.ext.types.dataclass
¶
A dataclass decorator which registers into optree's global namespace.