Space handling in equation parameter of tf.einsum (#19980)
* Fixes #19858 Adds space handling in the equation parameter similar to the np.einsum function to tf.einsum * Add tests for space handling Adds tests for the space handling in the equation parameter of tf.einsum and adjusts the `run_test` method to ignore the spaces when assigning dimensionality to the random input tensors for the tests.
Loading
Please sign in to comment