docstr: Checks support for additional dimensions in inputs.
docstr_tok: ["Checks","support","for","additional","dimensions","in","inputs","."]
code: def testHigherRankInputsForIW(self):
      """Checks support for additional dimensions in inputs."""
      placeholders = {
          "log_rhos": tf.placeholder(
              dtype=tf.float32, shape=[None, None, 1]
          ),
          "discounts": tf.placeholder(
              dtype=tf.float32, shape=[None, None, 1]
          ),
          "rewards": tf.placeholder(
              dtype=tf.float32, shape=[None, None, 42]
          ),
          "values": tf.placeholder(
              dtype=tf.float32, shape=[None, None, 42]
          ),
          "bootstrap_value": tf.placeholder(
              dtype=tf.float32, shape=[None, 42]
          ),
      }
      output = vtrace_ops.vtrace_from_importance_weights(**placeholders)
      self.assertEqual(output.vs.shape.as_list()[-1], 42)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  from trfl import vtrace_ops
label_desc: Use image node_id 3 for calling the VtraceTest obj's underlying member method code with example usage: obj.testHigherRankInputsForIW() without return types