head, attn = self.attention(qs, ks, vs, mask) heads.append(head); attns.append(attn) head = Concatenate()(heads) if n_head > 1 else heads[0]
確定! 回上一頁