Index

_

__call__() (geometric_algebra_attention.tensorflow.VectorAttention method)

B

build() (geometric_algebra_attention.keras.LabeledMultivectorAttention method) (geometric_algebra_attention.keras.LabeledVectorAttention method)

C

call() (geometric_algebra_attention.keras.Multivector2Vector method) (geometric_algebra_attention.keras.Vector2Multivector method) compute_mask() (geometric_algebra_attention.keras.LabeledMultivectorAttention method) (geometric_algebra_attention.keras.LabeledVectorAttention method) (geometric_algebra_attention.keras.TiedMultivectorAttention method) (geometric_algebra_attention.keras.TiedVectorAttention method)

F

forward() (geometric_algebra_attention.pytorch.Multivector2Vector method) (geometric_algebra_attention.pytorch.Vector2Multivector method)

from_config() (geometric_algebra_attention.keras.Multivector2MultivectorAttention class method) (geometric_algebra_attention.keras.Vector2VectorAttention class method)

G

geometric_algebra_attention.jax module geometric_algebra_attention.keras module geometric_algebra_attention.pytorch module

geometric_algebra_attention.tensorflow module get_config() (geometric_algebra_attention.keras.Multivector2MultivectorAttention method) (geometric_algebra_attention.keras.Vector2VectorAttention method)

L

LabeledMultivectorAttention (class in geometric_algebra_attention.jax) (class in geometric_algebra_attention.keras) (class in geometric_algebra_attention.pytorch) (class in geometric_algebra_attention.tensorflow)

LabeledVectorAttention (class in geometric_algebra_attention.jax) (class in geometric_algebra_attention.keras) (class in geometric_algebra_attention.pytorch) (class in geometric_algebra_attention.tensorflow)

M

module geometric_algebra_attention.jax geometric_algebra_attention.keras geometric_algebra_attention.pytorch geometric_algebra_attention.tensorflow MomentumLayerNormalization (class in geometric_algebra_attention.keras) (class in geometric_algebra_attention.pytorch) MomentumNormalization (class in geometric_algebra_attention.keras) (class in geometric_algebra_attention.pytorch) Multivector2MultivectorAttention (class in geometric_algebra_attention.jax) (class in geometric_algebra_attention.keras) (class in geometric_algebra_attention.pytorch) (class in geometric_algebra_attention.tensorflow)

Multivector2Vector (class in geometric_algebra_attention.jax) (class in geometric_algebra_attention.keras) (class in geometric_algebra_attention.pytorch) (class in geometric_algebra_attention.tensorflow) MultivectorAttention (class in geometric_algebra_attention.jax) (class in geometric_algebra_attention.keras) (class in geometric_algebra_attention.pytorch) (class in geometric_algebra_attention.tensorflow)

P

params (geometric_algebra_attention.jax.Multivector2MultivectorAttention property) (geometric_algebra_attention.jax.Vector2VectorAttention property)

S

scale_net (geometric_algebra_attention.jax.Multivector2MultivectorAttention property) (geometric_algebra_attention.jax.Vector2VectorAttention property)

stax_init() (geometric_algebra_attention.jax.Multivector2MultivectorAttention method) (geometric_algebra_attention.jax.Vector2VectorAttention method)

T

TiedMultivectorAttention (class in geometric_algebra_attention.jax) (class in geometric_algebra_attention.keras) (class in geometric_algebra_attention.pytorch) (class in geometric_algebra_attention.tensorflow)

TiedVectorAttention (class in geometric_algebra_attention.jax) (class in geometric_algebra_attention.keras) (class in geometric_algebra_attention.pytorch) (class in geometric_algebra_attention.tensorflow)

V

Vector2Multivector (class in geometric_algebra_attention.jax) (class in geometric_algebra_attention.keras) (class in geometric_algebra_attention.pytorch) (class in geometric_algebra_attention.tensorflow) Vector2VectorAttention (class in geometric_algebra_attention.jax) (class in geometric_algebra_attention.keras) (class in geometric_algebra_attention.pytorch) (class in geometric_algebra_attention.tensorflow)

VectorAttention (class in geometric_algebra_attention.jax) (class in geometric_algebra_attention.keras) (class in geometric_algebra_attention.pytorch) (class in geometric_algebra_attention.tensorflow)