Source code for geometric_algebra_attention.jax.TiedVectorAttention
from .. import base
from .Vector2VectorAttention import Vector2VectorAttention
[docs]
class TiedVectorAttention(base.TiedVectorAttention, Vector2VectorAttention):
__doc__ = base.TiedVectorAttention.__doc__