def __call__(self, c, a, b, s1, s2, z):
c, h = functions.lstm(
c,
self.w_az(a) + self.w_bz(b) + self.w_s1z(s1) + self.w_s2z(s2) + self.w_zz(z),
)
return c, XP.dropout(h)
微信
公众号
扫码关注公众号