Exemplo n.º 1
0
    def __init__(self,
                 in_channels,
                 kernel_size,
                 stride=1,
                 padding=0,
                 out_padding=0,
                 dilation=1,
                 bias=True,
                 multiplier=1,
                 weight_init=Kaiming_Normal(),
                 bias_init=Zeros()):
        """

        :param in_channels:
        :param kernel_size:
        :param stride:
        :param padding:
        :param dilation:
        :param bias:
        :param multiplier:
        :param weight_init:
        :param bias_init:
        """
        super(DepthwiseConvTranspose3d,
              self).__init__(in_channels, in_channels * multiplier,
                             _triple(kernel_size), _triple(stride),
                             _triple(padding),
                             _triple(dilation), in_channels, bias,
                             _triple(out_padding), weight_init, bias_init)
Exemplo n.º 2
0
    def __init__(self,
                 in_channels,
                 out_channels,
                 kernel_size,
                 stride=1,
                 padding=0,
                 out_padding=0,
                 dilation=1,
                 groups=1,
                 bias=True,
                 weight_init=Kaiming_Normal(),
                 bias_init=Zeros()):
        """

        :param in_channels:
        :param out_channels:
        :param kernel_size:
        :param stride:
        :param padding:
        :param dilation:
        :param groups:
        :param bias:
        :param weight_init:
        :param bias_init:
        """
        super(ConvTranspose3d,
              self).__init__(in_channels, out_channels, _triple(kernel_size),
                             _triple(stride), _triple(padding),
                             _triple(dilation), groups, bias,
                             _triple(out_padding), weight_init, bias_init)