Source code for geometric_algebra_attention.keras.Vector2Multivector


from tensorflow import keras

from .. import base
from .internal import AttentionBase



[docs]
class Vector2Multivector(base.Vector2Multivector, keras.layers.Layer):
    __doc__ = base.Vector2Multivector.__doc__

    math = AttentionBase.math

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



[docs]
    def call(self, inputs, mask=None):
        return self._evaluate(inputs)





keras.utils.get_custom_objects()['Vector2Multivector'] = Vector2Multivector