agedi.models.conditionings.scalar¶

Classes¶

ScalarConditioning

Conditioning module for continuous scalar properties.

Module Contents¶

class agedi.models.conditionings.scalar.ScalarConditioning(*args, input_dim: int = 1, output_dim: int = 2, **kwargs)¶

Bases: agedi.models.conditionings.base.Conditioning

Conditioning module for continuous scalar properties.

Projects a scalar property through a learned linear layer and encodes it with sinusoidal features (cos and sin), producing a 2-dimensional conditioning vector.

embedder¶

get_conditioning(x: torch.Tensor) → torch.Tensor¶

Get the conditioning tensor for x

Parameters:: x (torch.Tensor) – Time tensor of shape (Nodes, 1).
Returns:: Conditioning tensor of shape (Nodes, 2).
Return type:: torch.Tensor

get_empty_conditioning(n: int) → torch.Tensor¶

Get an empty conditioning tensor.

Returns:: Empty conditioning tensor of shape (n, 2).
Return type:: torch.Tensor