code: def test_2d(self, use_numpy):
      # np.shuffle only shuffles the first axis
      ax = tnp.asarray([[1, 2, 3], [4, 5, 6]])
      ox = ax.copy()
  
      tnp.random.seed(1234)
      tnp.random.shuffle(ax)
  
      assert isinstance(ax, tnp.ndarray)
      assert not (ax == ox).all()
  
imports: from contextlib import contextmanager
  from functools import partial
  import numpy
  import pytest
  import torch._dynamo.config
  import torch._numpy
  from torch._numpy.testing import assert_equal
  from torch.testing._internal.common_utils import instantiate_parametrized_tests, parametrize, run_tests, subtest, TestCase
label_desc: Use image node_id 2 for calling the TestShuffle obj's underlying member method code with example usage: obj.test_2d(use_numpy) without return types