code: def testHandlesExtraOuterDims(self):
      observation_spec = tensor_spec.BoundedTensorSpec(
          (8, 8, 3), tf.float32, 0, 1
      )
      observation = tensor_spec.sample_spec_nest(
          observation_spec, outer_dims=(3, 3, 2)
      )
  
      net = value_network.ValueNetwork(
          observation_spec,
          conv_layer_params=[(4, 2, 2)],
          fc_layer_params=(5,),
      )
  
      value, _ = net(observation)
      self.assertEqual([3, 3, 2], value.shape.as_list())
  
file_path: agents/tf_agents/networks/value_network_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  from tf_agents.networks import value_network
  from tf_agents.specs import tensor_spec
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 2 for calling the ValueNetworkTest obj's underlying member method code with example usage: obj.testHandlesExtraOuterDims() without return types