code: def setUp(self):
      self.k = 2
      self.batch_size = 64
      self.num_tokens = 10
      self.embedding_size = 768
      self.token_len = 10
      self.device = get_current_device()
      self.encoded_layers = [
          torch.randn(
              self.batch_size, self.token_len, self.embedding_size
          ).to(self.device)
          for _ in range(3)
      ]
      self.pad_mask = torch.randn(self.batch_size, self.token_len).to(
          self.device
      )
  
file_path: mmf/tests/modules/test_poolers.py
imports: import unittest
  import mmf.modules.poolers
  import torch
  from mmf.utils.general import get_current_device
label_desc: Use image node_id 1 for calling the TestModulePoolers obj's underlying member method code with example usage: obj.setUp() without return types