code: def testOutput(self):
      for use_gpu in [True, False]:
          with self.cached_session(use_gpu=use_gpu):
              logits, targets, pos_weight, losses = self._Inputs(
                  dtype=dtypes.float32
              )
              loss = nn_impl.weighted_cross_entropy_with_logits(
                  targets=targets, logits=logits, pos_weight=pos_weight
              )
              np_loss = np.array(losses).astype(np.float32)
              tf_loss = self.evaluate(loss)
          self.assertAllClose(np_loss, tf_loss, atol=0.001)
  
imports: import math
  import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import gradient_checker
  from tensorflow.python.ops import gradients_impl
  from tensorflow.python.ops import nn_impl
  import tensorflow.python.ops.nn_grad
  from tensorflow.python.platform import test
label_desc: Use image node_id 4 for calling the WeightedCrossEntropyTest obj's underlying member method code with example usage: obj.testOutput() without return types