具有张量流tf.matmul的3D张量

Question

我想用两个3-D张量进行乘法，如下定义：

a = tf.random_uniform(shape = [5,3,3])   

b = tf.ones(shape = [5,3,1])   

c = tf.matmul(a,b) 

但我无法得到正确的答案，如tf.matmul函数中所述

https://www.tensorflow.org/api_docs/python/tf/matmul

Answer 1

你确定你使用的是Tensorflow r1.0吗？

在旧版本中，您需要使用tf.batch_matmul来乘以批量矩阵。