def __call__(self, c, a, b, s1, s2, z):
c, h = functions.lstm(
c,
self.w_as(a) + self.w_bs(b) + self.w_s1s(s1) + self.w_s2s(s2) + self.w_zs(z),
)
return c, XP.dropout(h)
微信
公众号
扫码关注公众号