def __init__(self, d_model, d_ff, n_head, dropout):
super().__init__()
self.mh = MultiHeadAtt(n_head, d_model, dropout)
self.pw = PositionWise(d_model, d_ff, dropout)
微信
公众号
扫码关注公众号