code: def test_batch_shuffle(self, seed=0):
      torch.manual_seed(seed)
      x1 = torch.rand((4, 3, 64, 64))
      x1_shuffled, shuffle = batch_shuffle(x1)
      out1 = batch_unshuffle(x1_shuffled, shuffle)
      self.assertTrue(torch.equal(x1, out1))
      self.assertFalse(torch.equal(x1, x1_shuffled))
  
imports: import copy
  import unittest
  import torch
  import torch.nn
  from lightly.models import utils
  from lightly.models.utils import _no_grad_trunc_normal, activate_requires_grad, batch_shuffle, batch_unshuffle, deactivate_requires_grad, nearest_neighbors, normalize_weight, update_momentum
label_desc: Use image node_id 2 for calling the TestModelUtils obj's underlying member method code with example usage: obj.test_batch_shuffle(seed) without return types