aboutsummaryrefslogtreecommitdiffhomepage
path: root/.gitignore
diff options
context:
space:
mode:
authorGravatar Yong Tang <yong.tang.github@outlook.com>2017-08-07 15:50:27 -0700
committerGravatar Rasmus Munk Larsen <rmlarsen@google.com>2017-08-07 15:50:27 -0700
commit76bd2bcc639bbe85f9121894f2639324a004cc2c (patch)
treedab54b03485959dc61b2f3bbd52825b76abac0e4 /.gitignore
parent9c30ed2a7a424e8c05edc0e76224efe913a47077 (diff)
Fix tensordot with list of ints as axes (#11959)
* Fix tensordot with list of ints as axes This fix tries to fix the tensordot issue raised in 11950 where ``` TypeError: object of type 'int' has no len() ``` is thrown out when the axes is a list of ints: ``` with tf.Session() as sess: sess.run( tf.tensordot(tf.ones((3,3), dtype=tf.float32), tf.constant([2, 3, 1], dtype=tf.float32)[None, None], axes=[1, 2]) ) ``` In numpy it is possible to specify `np.tensordot(np.ones((3,3)), np.array([2, 3, 1])[None, None], axes=[1, 2])` This fix fixes #11950. Signed-off-by: Yong Tang <yong.tang.github@outlook.com> * Add test cases for tensordot with axes of list of ints This commit adds test cases for tensordot with axes of list of ints Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
Diffstat (limited to '.gitignore')
0 files changed, 0 insertions, 0 deletions