(dim_in, dim_out, kernel_size)
| 800 | |
| 801 | |
| 802 | def SameConv2d(dim_in, dim_out, kernel_size): |
| 803 | kernel_size = cast_tuple(kernel_size, 2) |
| 804 | padding = [k // 2 for k in kernel_size] |
| 805 | return nn.Conv2d(dim_in, dim_out, kernel_size=kernel_size, padding=padding) |
| 806 | |
| 807 | |
| 808 | class CausalConv3d(Module): |