(self, pad_mode: str = "reflect")
| 326 | """ |
| 327 | |
| 328 | def __init__(self, pad_mode: str = "reflect"): |
| 329 | super().__init__() |
| 330 | self.pad_mode = pad_mode |
| 331 | kernel_1d = torch.tensor([[1 / 8, 3 / 8, 3 / 8, 1 / 8]]) * 2 |
| 332 | self.pad = kernel_1d.shape[1] // 2 - 1 |
| 333 | self.register_buffer("kernel", kernel_1d.T @ kernel_1d, persistent=False) |
| 334 | |
| 335 | def forward(self, inputs: torch.Tensor) -> torch.Tensor: |
| 336 | inputs = F.pad(inputs, ((self.pad + 1) // 2,) * 4, self.pad_mode) |