code: def test_identity(self):
      pool_fn = poolers.IdentityPooler().to(self.device)
      out = pool_fn(self.encoded_layers[-1])
  
      assert (
          torch.Size(
              [self.batch_size, self.token_len, self.embedding_size]
          )
          == out.shape
      )
  
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 5 for calling the TestModulePoolers obj's underlying member method code with example usage: obj.test_identity() without return types