(inputs, kernel_size, dilation)
| 13 | |
| 14 | |
| 15 | def fixed_padding(inputs, kernel_size, dilation): |
| 16 | kernel_size_effective = kernel_size + (kernel_size - 1) * (dilation - 1) |
| 17 | pad_total = kernel_size_effective - 1 |
| 18 | pad_beg = pad_total // 2 |
| 19 | pad_end = pad_total - pad_beg |
| 20 | padded_inputs = F.pad(inputs, (pad_beg, pad_end, pad_beg, pad_end)) |
| 21 | return padded_inputs |
| 22 | |
| 23 | |
| 24 | class SeparableConv2d(nn.Module): |