ead Attention operation as defined in
[Attention Is All You Need](https://arxiv.org/abs/1706.03762) which takes
in a `query`, `key` and `value` tensors returns the dot-product attention
between them.</p>

    <h2 class="hasAnchor" id="examples"><a class="anchor" href="#examples"></a>Examples</h2>
    <pre class="examples"><div class='input'>
<span class='kw'>if</span> (<span class='fl'>FALSE</span>) {

<span class='no'>mha</span> <span class='kw'>=</span> <span class='fu'>layer_multi_head_attention</span>(<