modelzoo.vision.pytorch.dit.layers.TimestepEmbeddingLayer

Classes

TimestepEmbeddingLayer

Embeds scalar timesteps into vector representations.

previous

modelzoo.vision.pytorch.dit.layers.RegressionHead.RegressionHead

next

modelzoo.vision.pytorch.dit.layers.TimestepEmbeddingLayer.TimestepEmbeddingLayer