qrnn.py 文件源码

def __init__(self, in_channels, out_channels, kernel_size=2, pooling="f", zoneout=False, wgain=1., weightnorm=False):
        super(QRNNDecoder, self).__init__(in_channels, out_channels, kernel_size, pooling, zoneout, wgain, weightnorm)
        self.num_split = len(pooling) + 1
        wstd = math.sqrt(wgain / in_channels / kernel_size)

        with self.init_scope():
            setattr(self, "V", links.Linear(out_channels, self.num_split * out_channels, initialW=initializers.Normal(wstd)))

    # ht_enc is the last encoder state