docstr: Forward.
  
  Args:
      x: [M, B, K]
  
  Returns:
      [M, B, K]
docstr_tok: ["Forward",".","Args",":","x",":","[","M",",","B",",","K","]","Returns",":","[","M",",","B",",","K","]"]
code: def forward(self, x):
      """Forward.
  
      Args:
          x: [M, B, K]
  
      Returns:
          [M, B, K]
      """
      residual = x
      out = self.net(x)
      # TODO(Jing): when P = 3 here works fine, but when P = 2 maybe need to pad?
      return out + residual
  
imports: import torch
  import torch.nn
label_desc: Use image node_id 2 for calling the TemporalBlock obj's underlying member method code with example usage: obj.forward(x) and returns: unknown