trajdl.algorithms.embeddings.ctle module

class trajdl.algorithms.embeddings.ctle.CTLETokenEmbedding(embedding_type: str, tokenizer: AbstractTokenizer, embedding_dim: int, max_len: int)

Bases: BaseTokenEmbeddingLayer

property embedding_type: str

forward(masked_tokens: LongTensor, ts_src: LongTensor)

masked_tokens: shape is (B, T) ts_src: shape is (B, T)

class trajdl.algorithms.embeddings.ctle.CTLETokenEmbeddingWithTransformer(embedding_type: str, embedding_dim: int, max_len: int, num_layers: int, n_heads: int, tokenizer: str | AbstractTokenizer, hidden_size: int, dropout: float = 0.1)

Bases: BaseTokenEmbeddingLayer

forward(loc_src: LongTensor, ts_src: LongTensor, mask: BoolTensor)

Must override in subclass to compute embeddings.

Parameters:

x (torch.LongTensor) – Input tensor containing token indices.

Returns:

Embedding tensor for the input tokens, with increased dimensions.

Return type:

torch.Tensor

class trajdl.algorithms.embeddings.ctle.PositionalEncoding(embedding_dim: int, max_len: int)

Bases: Module

forward(x)

shape of x: (B, T, *)

class trajdl.algorithms.embeddings.ctle.TemporalEncoding(embedding_dim: int)

Bases: Module

forward(x: LongTensor)

x: shape is (B, T)