/td>
    </tr>
    <tr>
      <th>dropout</th>
      <td><p>float, `rate` parameter for the dropout layer that is applied to attention after softmax, default `0`.</p></td>
    </tr>
    <tr>
      <th>use_projection_bias</th>
      <td><p>bool, whether to use a bias term after the linear output projection.</p></td>
    </tr>
    <tr>
      <th>return_attn_coef</th>
      <td><p>bool, if `TRUE`, return the attention coefficients as an additional output argument.</p></td>
    </tr>
    <tr>
      <th>kernel_i