我如何在张量流中使用多头注意力？

Question

我想在tensorflow中使用多头注意力，我想知道在tensorflow中是否有使用多头注意力的api或有人在github上构建？