code: def forward(self, x):
      t, n = x.size(0), x.size(1)
      x = x.view(t * n, -1)
      x = self.module(x)
      x = x.view(t, n, -1)
      return x
  
imports: import math
  from collections import OrderedDict
  from typing import Optional, Tuple
  import torch
  import torch.nn.functional
  from torch import nn, Tensor
label_desc: Use image node_id 2 for calling the SequenceWise obj's underlying member method code with example usage: obj.forward(x) and returns: x