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

Question

I find the following on the Tensorflow documentation homepage for using the matmul operation when rank>2:

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]]]

It simply isn't working when I plug it in Python. I get

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

Anyone know what is wrong?

Answer 1

Is your TensorFlow too old? Here's what I get in version 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)