code: def testBatched(self):
      # Given batched action values:
      action_values = [
          [1.0, 2.0, 3.0],
          [4.0, 5.0, 6.0],
          [6.0, 5.0, 4.0],
          [3.0, 2.0, 1.0],
      ]
      epsilon = 0.0
  
      # We expect batched probabilities.
      expected = [
          [0.0, 0.0, 1.0],
          [0.0, 0.0, 1.0],
          [1.0, 0.0, 0.0],
          [1.0, 0.0, 0.0],
      ]
  
      result = policy_ops.epsilon_greedy(action_values, epsilon).probs
      with self.test_session() as sess:
          self.assertAllClose(sess.run(result), expected)
  
file_path: trfl/trfl/policy_ops_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow.compat.v1
  from trfl import policy_ops
label_desc: Use image node_id 4 for calling the EpsilonGreedyTest obj's underlying member method code with example usage: obj.testBatched() without return types