docstr: Test that the jacobian is correct.
code: def test_unflatten_batch_to_2d_jacobian_random(
      self, sizes, max_rows, num_features
  ):
      """Test that the jacobian is correct."""
      max_rows = np.max(sizes) if max_rows is None else max_rows
      total_rows = np.sum(sizes)
      data_init = 0.1 + np.random.uniform(
          size=(total_rows, num_features)
      )
  
      def unflatten_2d_to_batch(data):
          return utils.unflatten_2d_to_batch(data, sizes, max_rows)
  
      self.assert_jacobian_is_correct_fn(
          unflatten_2d_to_batch, [data_init]
      )
  
imports: from absl.testing import parameterized
  import numpy
  from scipy import linalg
  import tensorflow
  from tensorflow_graphics.geometry.convolution import utils
  from tensorflow_graphics.util import test_case
label_desc: Use image node_id 5 for calling the UtilsUnflatten2dToBatchTest obj's underlying member method code with example usage: obj.test_unflatten_batch_to_2d_jacobian_random(sizes, max_rows, num_features) and returns: utils