network_pytorch.py 文件源码

def atrous_conv(self,
                    input,
                    k_h, 
                    k_w, 
                    c_o, 
                    dilation,
                    name,
                    relu=True,
                    padding=DEFAULT_PADDING,
                    group=1,
                    biased=True):
        if group != 1:
            raise NotImplementedError
        input = pad_if_needed(input, padding, 'atrous_conv2d', k_h, k_w, dilation=dilation)

        result = F.conv2d(input, 
                          self.weights[name + '/weights'], 
                          bias=self.weights[name + '/biases'] if biased else None,
                          padding=0, 
                          dilation=dilation,
                          groups=group,
                          stride=1)
        if relu:
            result = F.relu(result)
        return result