code: def test_scale(self, t_, scale):
      t = tf.Variable([t_])
      with tf.GradientTape() as tape:
          y = scale_gradient.scale_gradient(t, scale)
          output = y * y
      grad = tape.gradient(output, t)
      self.assertAllEqual(grad.numpy(), [2 * t_ * scale])
      self.assertAllEqual(output.numpy(), [t_**2])
  
file_path: sonnet/sonnet/src/scale_gradient_test.py
imports: import itertools
  from absl.testing import parameterized
  from sonnet.src import scale_gradient
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 1 for calling the ScaleGradientTest obj's underlying member method code with example usage: obj.test_scale(t_, scale) without return types