class_docstr: Test cases for einsum op.
code: def testUnary(self):
      for dtype in self.float_types:
          self._testUnary(
              lambda x: special_math_ops.einsum("ijk->kji", x),
              np.array([[[1, 3], [2, 5], [6, 8]]], dtype=dtype),
              expected=np.array(
                  [[[1], [2], [6]], [[3], [5], [8]]], dtype=dtype
              ),
          )
  
file_path: tensorflow-upstream/tensorflow/compiler/tests/einsum_op_test.py
imports: import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import special_math_ops
  from tensorflow.python.platform import googletest
label_desc: Use image node_id 6 for calling the EinsumOpTest obj's underlying member method code with example usage: obj.testUnary() without return types