class_docstr: Test the `pixel_control_loss` op.
class_docstr_tok: ["Test","the","`","pixel_control_loss","`","op","."]
code: def testTensorDiscountShape(self):
      with self.assertRaisesRegexp(
          ValueError,
          "discount_factor must be a scalar or a tensor of rank 2",
      ):
          tensor_discount = tf.tile(
              tf.reshape(self.discount, [1, 1, 1]),
              [self.seq_length, self.batch_size, 1],
          )
          pixel_control_ops.pixel_control_loss(
              self.observations_ph,
              self.actions_ph,
              self.action_values_ph,
              self.cell_size,
              tensor_discount,
              self.scale,
          )
  
file_path: trfl/trfl/pixel_control_ops_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  import tensorflow.compat.v1
  from trfl import pixel_control_ops
label_desc: Use image node_id 5 for calling the PixelControlLossTest obj's underlying member method code with example usage: obj.testTensorDiscountShape() without return types