| 59 | |
| 60 | |
| 61 | class IdentityTransform(AbstractTransform): |
| 62 | def __init__(self, exp_type: str): |
| 63 | super().__init__(exp_type) |
| 64 | self._out_dim = self.input_dim |
| 65 | |
| 66 | def transform(self, X_not_batched: Dict[str, np.ndarray]) -> Dict[str, np.ndarray]: |
| 67 | return X_not_batched |
| 68 | |
| 69 | def batched_transform(self, X: Dict[str, np.ndarray]) -> Dict[str, np.ndarray]: |
| 70 | return X |
| 71 | |
| 72 | |
| 73 | class FlattenTransform(AbstractTransform): |