code: def test_big_prime_number_matmul(self):
      a = Tensor.rand(367, 367)
      b = Tensor.rand(367, 367)
      c = (a @ b).realize()
      np.testing.assert_allclose(
          c.numpy(), a.numpy() @ b.numpy(), atol=1e-4, rtol=1e-4
      )
  
file_path: tinygrad/extra/optimization/test_beam_search.py
imports: import unittest
  import numpy
  from tinygrad.helpers import BEAM, Timing
  from tinygrad.shape.symbolic import Variable
  from tinygrad.tensor import Tensor
  from tinygrad.nn import Conv2d
label_desc: Use image node_id 4 for calling the TestBeamSearch obj's underlying member method code with example usage: obj.test_big_prime_number_matmul() without return types