这两个
tf.tensordot()和
tf.einsum()是语法糖那套一个或更多的调用
tf.matmul()(虽然在一些特殊情况下
tf.einsum()可以降低到简单的按元素
tf.multiply())。
在极限情况下,我希望所有三个函数在同一计算中都具有相同的性能。但是,对于较小的矩阵,
tf.matmul()直接使用可能会更有效,因为它会产生带有较少操作的更简单的TensorFlow图,因此每操作调用的成本会更低。

这两个
tf.tensordot()和
tf.einsum()是语法糖那套一个或更多的调用
tf.matmul()(虽然在一些特殊情况下
tf.einsum()可以降低到简单的按元素
tf.multiply())。
在极限情况下,我希望所有三个函数在同一计算中都具有相同的性能。但是,对于较小的矩阵,
tf.matmul()直接使用可能会更有效,因为它会产生带有较少操作的更简单的TensorFlow图,因此每操作调用的成本会更低。