class_docstr: A residual connection followed by a layer norm.
  
  Args:
      size (int): layerNorm size
      dropout (float): the dropout probability
      device: (Optional[torch.device]).
  
  Example::
  
      self.input_sublayer = SublayerConnection(
          size=hidden, dropout=dropout, device=device
      )
class_docstr_tok: ["A","residual","connection","followed","by","a","layer","norm",".","Args",":","size","(","int",")",":","layerNorm","size","dropout","(","float",")",":","the","dropout","probability","device",":","(","Optional","[","torch.device","]",")",".","Example",":",":","self.input_sublayer","=","SublayerConnection","(","size=hidden",",","dropout=dropout",",","device=device",")"]
docstr: forward function including the norm layer, sublayer and dropout, finally
  add up with the input tensor
  
  Args:
      x (torch.Tensor): the input tensor
      sublayer (Callable[[torch.Tensor): callable layer
  
  Returns:
      torch.Tensor.
docstr_tok: ["forward","function","including","the","norm","layer",",","sublayer","and","dropout",",","finally","add","up","with","the","input","tensor","Args",":","x","(","torch.Tensor",")",":","the","input","tensor","sublayer","(","Callable","[","[","torch.Tensor",")",":","callable","layer","Returns",":","torch.Tensor","."]
code: def forward(
      self,
      x: torch.Tensor,
      sublayer: Callable[[torch.Tensor], torch.Tensor],
  ) -> torch.Tensor:
      """
      forward function including the norm layer, sublayer and dropout, finally
      add up with the input tensor
  
      Args:
          x (torch.Tensor): the input tensor
          sublayer (Callable[[torch.Tensor): callable layer
  
      Returns:
          torch.Tensor.
      """
      return x + self.dropout(sublayer(self.norm(x)))
  
imports: import copy
  import math
  from typing import Callable, Optional, Tuple
  import torch
  import torch.nn
  from torchrec.modules.embedding_configs import EmbeddingConfig
  from torchrec.modules.embedding_modules import EmbeddingCollection
  from torchrec.sparse.jagged_tensor import KeyedJaggedTensor
label_desc: Use image node_id 2 for calling the SublayerConnection obj's underlying member method code with example usage: obj.forward(x, sublayer) and returns: unknown