MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / testUnary

Method testUnary

tensorflow/python/kernel_tests/einsum_op_test.py:53–67  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

51 self.assertAllClose(a, b, atol=1e-4, rtol=1e-4)
52
53 def testUnary(self):
54 self._check('->', ())
55 self._check('aa->', (3, 3))
56 self._check('aa->a', (3, 3))
57 self._check('aaa->', (3, 3, 3))
58 self._check('aaa->a', (3, 3, 3))
59 self._check('aab->a', (3, 3, 4))
60 self._check('ab->', (3, 3))
61 self._check('ab->ab', (3, 3))
62 self._check('abc->b', (3, 4, 5))
63 self._check('abc->ca', (3, 4, 5))
64 self._check('abc->cab', (3, 4, 5))
65 self._check('aabcc->a', (3, 3, 5, 4, 4))
66 self._check('aabcc->ac', (3, 3, 5, 4, 4))
67 self._check('aabcd->ad', (3, 3, 5, 4, 4))
68
69 def testUnaryEllipsis(self):
70 self._check('...->...', ())

Callers

nothing calls this directly

Calls 1

_checkMethod · 0.95

Tested by

no test coverage detected