卷积层 此卷积非彼卷积,实际上为二维信号的互相关运算,使用scipy.signal.correlate(x, kernel, ‘valid’)进行运算 from util_pkg import * from scipy import signal import torch from torch import nn class Conv2D(nn.Module): def __init__(self, kernel_size): super().__init__() self.kernel = nn.Parameter(torch.randn(kernel
