Return the masked tensor along with optional loss/metrics.
(self, tensor: Tensor, step: int)
| 25 | """Observe the current splats before running compression.""" |
| 26 | |
| 27 | def apply(self, tensor: Tensor, step: int) -> MaskResult: |
| 28 | """Return the masked tensor along with optional loss/metrics.""" |
| 29 | return MaskResult(value=tensor) |
| 30 | |
| 31 | def step_optimizer(self, step: int) -> None: |
| 32 | """Hook for stepping any optimizers the mask may own.""" |