| 13 | |
| 14 | |
| 15 | class MappingL2(Mapping[MappingL2Cfg]): |
| 16 | def forward_undistorted( |
| 17 | self, |
| 18 | delta: Float[Tensor, "*batch 2"], |
| 19 | ) -> Float[Tensor, " *batch"]: |
| 20 | # Multiply by 0.5 to match torch.nn.functional.huber_loss. |
| 21 | return 0.5 * (delta * delta).sum(dim=-1) |
nothing calls this directly
no outgoing calls
no test coverage detected