utils module

Full Documentation for hippynn.optimizer.utils module. Click here for a summary page.

Internal utilities for batch optimizers

debatch(numbers: Tensor, coords: Tensor, numbers_padding: int = 0, return_mask: bool = False) → List[Tensor] | Tuple[List[Tensor], Tensor]

debatch_coords(coords: Tensor, masks: Tensor) → List[Tensor]

debatch_numbers(numbers: Tensor, padding_number: int = 0, return_mask: bool = False) → List[Tensor] | Tuple[List[Tensor], Tensor]