code: def init_weights(self, gain: float = 1.0):
      torch.nn.init.normal_(
          self.position_embeddings.weight, std=0.02 * gain
      )
      torch.nn.init.normal_(
          self.word_embeddings.weight, std=0.02 * gain
      )
  
file_path: xformers/xformers/components/positional_embedding/vocab.py
imports: from dataclasses import dataclass
  from typing import Optional
  import torch
  import torch.nn
  from xformers.components.positional_embedding import PositionEmbedding, PositionEmbeddingConfig, register_positional_embedding
label_desc: Use image node_id 2 for calling the VocabEmbedding obj's underlying member method code with example usage: obj.init_weights(gain) without return types