quaterion_models.types package

Module contents

CollateFnType

alias of Callable[[List[Any]], Union[Tensor, Tuple[Tensor], List[Tensor], Dict[str, Tensor], Dict[str, dict], Any]]

TensorInterchange

alias of Union[Tensor, Tuple[Tensor], List[Tensor], Dict[str, Tensor], Dict[str, dict], Any]