def __init__(self, insize, outsize):
self.insize = insize
self.outsize = outsize
self.h = None
self.c = None
self.peep_dim = insize + 2 * outsize
super(DyerLSTM, self).__init__(
linear_in=F.Linear(self.peep_dim, outsize, bias=0.25,
wscale=_dyer_init(self.peep_dim, outsize)),
linear_c=F.Linear(insize + outsize, outsize,
wscale=_dyer_init(insize + outsize, outsize)),
linear_out=F.Linear(self.peep_dim, outsize,
wscale=_dyer_init(self.peep_dim, outsize))
)
评论列表
文章目录