Tensorflow中tf.matmul和tf.batch_matmul有什么区别?

时间:2017-01-30 15:57:34

标签: tensorflow

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

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)

1 个答案:

答案 0 :(得分:3)

在版本0.12及更高版本中,不推荐使用

tf.batch_matmul以支持tf.matmul,因此在以后的版本中没有区别。早期版本需要matmul的排名2输入,但允许batch_matmul

的排名更高