使用Numpy/TensorFlow中的tensordot进行多维矩阵相乘
在使用tensordot进行矩阵相乘的时候,axes是一个重要的参数,它可以定义两个矩阵互乘的时候是使用内积、外积、或是其他方法。
以TensorFlow的tensordot为例,其语法为:
tf.tensordot(
a, b, axes, name=None
)
其中a和b为输入的两个矩阵,参数axes定义了两个矩阵如何相乘。
如设置“axes=0”的时候,相当于定义为算a和...
原创
2020-04-14 19:08:28 ·
2608 阅读 ·
0 评论