code: def get_predictions(self, num_decodes=2):
      rng = np.random.RandomState(0)
      # num_samples=4
      inputs = rng.randint(0, 255, (4, 2, 64, 64, 3))
      outputs = rng.randint(0, 255, (4, 5, 64, 64, 3))
      targets = rng.randint(0, 255, (4, 5, 64, 64, 3))
      predictions = []
      for input_, output, target in zip(inputs, outputs, targets):
          curr_pred = {
              "inputs": input_,
              "outputs": output,
              "targets": target,
          }
          predictions.append(curr_pred)
  
      # num_decodes=2
      predictions = [predictions] * num_decodes
      problem = registry.problem("video_stochastic_shapes10k")
      return predictions, problem
  
file_path: tensor2tensor/tensor2tensor/data_generators/video_utils_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import numpy
  from tensor2tensor.data_generators import video_generated
  from tensor2tensor.data_generators import video_utils
  from tensor2tensor.utils import decoding
  from tensor2tensor.utils import registry
  import tensorflow.compat.v1
label_desc: Use image node_id 1 for calling the VideoUtilsTest obj's underlying member method code with example usage: obj.get_predictions(num_decodes) and returns: predictions,  problem