jax.nn.dot\_product\_attention
==============================

.. currentmodule:: jax.nn

.. autofunction:: dot_product_attention