layers.continuous_sincos_embedding_config

Classes

ContinuousSincosEmbeddingConfig

Configuration for Continuous Sine-Cosine Embedding layer.

Module Contents

class layers.continuous_sincos_embedding_config.ContinuousSincosEmbeddingConfig(/, **data)

Bases: pydantic.BaseModel

Configuration for Continuous Sine-Cosine Embedding layer.

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:

data (Any)

hidden_dim: int = None

Dimensionality of the output embedding.

input_dim: int = None

Dimensionality of the input coordinates.

max_wavelength: int = None

Maximum wavelength for the sine-cosine embeddings.

assert_positive: bool = None

Whether to assert that all input coordinates are positive.