code: def __init__(self, emb, depth, lr_mul=0.1):
      super().__init__()
  
      layers = []
      for _ in range(depth):
          layers.extend(
              [
                  EqualLinear(emb, emb, lr_mul),
                  torch.nn.LeakyReLU(0.2, inplace=True),
              ]
          )
  
      self.net = torch.nn.Sequential(*layers)
  
imports: import math
  from functools import partial
  from math import log2
  from typing import List
  import torch
  import torch.nn.functional
  from einops import rearrange
  from kornia.filters import filter2d
  from nemo.collections.tts.parts.utils.helpers import mask_sequence_tensor
label_desc: Use image node_id 1 to create a new StyleMapping object from inherited base classes: torch.nn with example: obj = StyleMapping(emb, depth, lr_mul)