class_docstr: Logical "Tower" of embeddings directly passed to provided interaction.
  All TorchRec shardable embedding modules are supported.
  
  Args:
      embedding_module (nn.Module):
      interaction_module (nn.Module):
      device (Optional[torch.device]):
  
  Example::
  
      ebc, interaction = EmbeddingBagCollection(), MyInteractionModule()
      tower = EmbeddingTower(ebc, interaction, device)
      kjt = KeyedJaggedTensor()
      output = tower(kjt)
docstr: Executes the embedding module and interaction module.
  
  Args:
      *args (Any): user provided positional arguments.
      **kwargs (Any): user provided keyword arguments.
  
  Returns:
      torch.Tensor: 2-D tensor of shape of `N X B`, where `B` is local batch size.
docstr_tok: ["Executes","the","embedding","module","and","interaction","module",".","Args",":","*","args","(","Any",")",":","user","provided","positional","arguments",".","*","*","kwargs","(","Any",")",":","user","provided","keyword","arguments",".","Returns",":","torch.Tensor",":","2-D","tensor","of","shape","of","`","N","X","B","`",",","where","`","B","`","is","local","batch","size","."]
code: def forward(
      self,
      # pyre-ignore [2]
      *args,
      # pyre-ignore [2]
      **kwargs,
  ) -> torch.Tensor:
      """
      Executes the embedding module and interaction module.
  
      Args:
          *args (Any): user provided positional arguments.
          **kwargs (Any): user provided keyword arguments.
  
      Returns:
          torch.Tensor: 2-D tensor of shape of `N X B`, where `B` is local batch size.
      """
      embeddings = self.embedding(*args, **kwargs)
      return self.interaction(embeddings)
  
file_path: torchrec/torchrec/modules/embedding_tower.py
imports: from typing import List, Optional, Tuple
  import torch
  import torch.nn
  from torchrec.modules.embedding_modules import EmbeddingBagCollection, EmbeddingCollection
  from torchrec.sparse.jagged_tensor import KeyedJaggedTensor
label_desc: Use image node_id 2 for calling the EmbeddingTower obj's underlying member method code with example usage: obj.forward() and returns: self