例如,如果我有以下数据:
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)
答案 0 :(得分:3)
tf.batch_matmul
以支持tf.matmul
,因此在以后的版本中没有区别。早期版本需要matmul
的排名2输入,但允许batch_matmul