Source code for geometric_algebra_attention.pytorch.LabeledVectorAttention


from .. import base
from .Vector2VectorAttention import Vector2VectorAttention


[docs]
class LabeledVectorAttention(base.LabeledVectorAttention, Vector2VectorAttention):
    __doc__ = base.LabeledVectorAttention.__doc__

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        if type(self) == LabeledVectorAttention:
            self.init()