diff options
| author |  Jacques Pienaar <jpienaar@google.com> | 2018-03-21 12:07:51 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-21 12:10:30 -0700 |
| commit | 2d0531d72c7dcbb0e149cafdd3a16ee8c3ff357a (patch) | |
| tree | 1179ecdd684d10c6549f85aa95f33dd79463a093 /tensorflow/python/ops/special_math_ops_test.py | |
| parent | cbede3ea7574b36f429710bc08617d08455bcc21 (diff) | |
Merge changes from github.
PiperOrigin-RevId: 189945839
Diffstat (limited to 'tensorflow/python/ops/special_math_ops_test.py')
| -rw-r--r-- | tensorflow/python/ops/special_math_ops_test.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/python/ops/special_math_ops_test.py b/tensorflow/python/ops/special_math_ops_test.py index 2c212f4548..d7c3a7e8dc 100644 --- a/tensorflow/python/ops/special_math_ops_test.py +++ b/tensorflow/python/ops/special_math_ops_test.py @@ -192,6 +192,9 @@ class EinsumTest(test.TestCase): 'abc,cba', 'dba,ead,cad->bce', 'aef,fbc,dca->bde', + 'iJ,Jk->ik', + 'iJ,Ki->JK', + 'iJk,Jklm->Jk' ] long_cases = [ @@ -208,6 +211,8 @@ class EinsumTest(test.TestCase): 'ijk ijk', 'ij.jk->ik', 'ij...,jk...->ik...', + 'ij,k ->kji', + 'ij,k-> kji', # axis in output that does not exist 'ij,jk->im', |