| 106 | |
| 107 | |
| 108 | class Blur(nn.Module): |
| 109 | def __init__(self, kernel, pad, upsample_factor=1): |
| 110 | super().__init__() |
| 111 | |
| 112 | kernel = make_kernel(kernel) |
| 113 | |
| 114 | if upsample_factor > 1: |
| 115 | kernel = kernel * (upsample_factor ** 2) |
| 116 | |
| 117 | self.register_buffer('kernel', kernel) |
| 118 | |
| 119 | self.pad = pad |
| 120 | |
| 121 | def forward(self, input): |
| 122 | return upfirdn2d(input, self.kernel, pad=self.pad) |
| 123 | |
| 124 | |
| 125 | class EqualConv2d(nn.Module): |