等级> 2的Tensorflow matmul操作不起作用

Question

我在Tensorflow文档主页上找到以下内容，以便在排名＆gt; 2时使用matmul操作：

https://www.tensorflow.org/api_docs/python/math_ops/matrix_math_functions#matmul

# 3-D tensor `a`
a = tf.constant(np.arange(1,13), shape=[2, 2, 3]) => [[[ 1.  2.  3.]
                                                   [ 4.  5.  6.]],
                                                  [[ 7.  8.  9.]
                                                   [10. 11. 12.]]]

# 3-D tensor `b`
b = tf.constant(np.arange(13,25), shape=[2, 3, 2]) => [[[13. 14.]
                                                    [15. 16.]
                                                    [17. 18.]],
                                                   [[19. 20.]
                                                    [21. 22.]
                                                    [23. 24.]]]
c = tf.matmul(a, b) => [[[ 94 100]
                     [229 244]],
                    [[508 532]
                     [697 730]]]

当我用Python插入它时，它根本不起作用。我得到了

c = tf.matmul(a, b)
ValueError: Shape must be rank 2 but is rank 3

任何人都知道出了什么问题？

Answer 1

你的TensorFlow太旧了吗？这是我在0.12rc0版本中获得的内容

a = tf.constant(np.arange(1,13).astype(np.float32), shape=[2, 2, 3])
b = tf.constant(np.arange(13,25).astype(np.float32), shape=[2, 3, 2])
sess.run(tf.matmul(a, b)) =>

array([[[  94.,  100.],
        [ 229.,  244.]],

       [[ 508.,  532.],
        [ 697.,  730.]]], dtype=float32)