File size: 353 Bytes
002ca81 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from torch import nn
def Conv2dSame(dim_in, dim_out, kernel_size, bias=True):
pad_left = kernel_size // 2
pad_right = (pad_left - 1) if (kernel_size % 2) == 0 else pad_left
return nn.Sequential(
nn.ZeroPad2d((pad_left, pad_right, pad_left, pad_right)),
nn.Conv2d(dim_in, dim_out, kernel_size, bias=bias),
)
|