ze, query_elements, query_depth)</span>
<span class='no'>key</span> <span class='kw'>=</span> <span class='no'>tf</span>$<span class='no'>random</span>$<span class='fu'>uniform</span>(<span class='fu'><a href='https://rdrr.io/r/base/list.html'>list</a></span>(<span class='fl'>32L</span>, <span class='fl'>15L</span>, <span class='fl'>300L</span>)) <span class='co'># (batch_size, key_elements, key_depth)</span>
<span class='no'>value</span> <span class='kw'>=</span> <span class='no'>tf</span>$<span class='no'