    <th>object</th>
      <td><p>Model or layer object</p></td>
    </tr>
    <tr>
      <th>head_size</th>
      <td><p>int, dimensionality of the `query`, `key` and `value` tensors after the linear transformation.</p></td>
    </tr>
    <tr>
      <th>num_heads</th>
      <td><p>int, number of attention heads.</p></td>
    </tr>
    <tr>
      <th>output_size</th>
      <td><p>int, dimensionality of the output space, if `NULL` then the input dimension of `value` or `key` will be used, default `NULL`.</p><