torchcell.models.diffusion_decoder.SinusoidalTimeEmbedding

class SinusoidalTimeEmbedding(dim: int)

Bases: Module

Sinusoidal time embeddings for diffusion timesteps.

forward(timesteps: Tensor) → Tensor

Embed diffusion timesteps as sinusoidal features.

Parameters:

timesteps – [batch_size] tensor of timesteps

Returns:

[batch_size, dim] tensor of time embeddings