tensorflow - Tensorflow 中的 tf.matmul 和 tf.batch_matmul 有什么区别？

Question

例如，如果我有以下数据:

x = tf.placeholder("float", [None, n, n])
y = tf.placeholder("float", [None, n, n])

这两种操作有什么区别吗？

res = tf.matmul(x,y)
res = tf.batch_matmul(x,y)

Answer 1

tf.batch_matmul不赞成使用 tf.matmul在 0.12 及更高版本中，因此在更高版本中没有区别。早期版本需要 matmul 的 rank-2 输入但允许更大的排名 batch_matmul