class_docstr: Norm layer modified to incorporate timestep embeddings.
  
  Parameters:
      embedding_dim (`int`): The size of each embedding vector.
      num_embeddings (`int`): The size of the embeddings dictionary.
class_docstr_tok: ["Norm","layer","modified","to","incorporate","timestep","embeddings",".","Parameters",":","embedding_dim","(","`","int","`",")",":","The","size","of","each","embedding","vector",".","num_embeddings","(","`","int","`",")",":","The","size","of","the","embeddings","dictionary","."]
code: def forward(
      self, x: torch.Tensor, timestep: torch.Tensor
  ) -> torch.Tensor:
      emb = self.linear(self.silu(self.emb(timestep)))
      scale, shift = torch.chunk(emb, 2)
      x = self.norm(x) * (1 + scale) + shift
      return x
  
imports: import numbers
  from typing import Dict, Optional, Tuple
  import torch
  import torch.nn
  import torch.nn.functional
  from ..utils import is_torch_version
  from .activations import get_activation
  from .embeddings import CombinedTimestepLabelEmbeddings, PixArtAlphaCombinedTimestepSizeEmbeddings
label_desc: Use image node_id 2 for calling the AdaLayerNorm obj's underlying member method code with example usage: obj.forward(x, timestep) and returns: x