(self, pad_mode: str = "reflect")
| 261 | """ |
| 262 | |
| 263 | def __init__(self, pad_mode: str = "reflect"): |
| 264 | super().__init__() |
| 265 | self.pad_mode = pad_mode |
| 266 | kernel_1d = torch.tensor([[1 / 8, 3 / 8, 3 / 8, 1 / 8]]) |
| 267 | self.pad = kernel_1d.shape[1] // 2 - 1 |
| 268 | self.register_buffer("kernel", kernel_1d.T @ kernel_1d, persistent=False) |
| 269 | |
| 270 | def forward(self, inputs: torch.Tensor) -> torch.Tensor: |
| 271 | inputs = F.pad(inputs, (self.pad,) * 4, self.pad_mode) |