(self, patch_size: Sequence[int] | int, device: torch.device | str | None = None)
| 39 | """ |
| 40 | |
| 41 | def __init__(self, patch_size: Sequence[int] | int, device: torch.device | str | None = None) -> None: |
| 42 | self.patch_size = patch_size |
| 43 | self.device = device |
| 44 | |
| 45 | @abstractmethod |
| 46 | def get_input_shape(self, inputs: Any) -> tuple: |