(self, pad_mode: str = "reflect")
| 332 | """ |
| 333 | |
| 334 | def __init__(self, pad_mode: str = "reflect"): |
| 335 | super().__init__() |
| 336 | self.pad_mode = pad_mode |
| 337 | kernel_1d = torch.tensor([[1 / 8, 3 / 8, 3 / 8, 1 / 8]]) * 2 |
| 338 | self.pad = kernel_1d.shape[1] // 2 - 1 |
| 339 | self.register_buffer("kernel", kernel_1d.T @ kernel_1d, persistent=False) |
| 340 | |
| 341 | def forward(self, inputs: torch.Tensor) -> torch.Tensor: |
| 342 | inputs = F.pad(inputs, ((self.pad + 1) // 2,) * 4, self.pad_mode) |