tensorflow--math--3--矩阵乘法 ( tf.matmul )

tf.matmul

用来做矩阵乘法。若a为l*m的矩阵,b为m*n的矩阵,那么通过tf.matmul(a,b) 结果就会得到一个l*n的矩阵 不过这个函数还提供了很多额外的功能。我们来看下函数的定义:

matmul(a, b,
           transpose_a=False, transpose_b=False,
           a_is_sparse=False, b_is_sparse=False,
           name=None):

可以看到还提供了transpose和is_sparse的选项。

如果对应的transpose项为True,例如transpose_a=True,那么a在参与运算之前就会先转置一下。

而如果a_is_sparse=True,那么a会被当做稀疏矩阵来参与运算。

个人公众号,比较懒,很少更新,可以在上面提问题,如果回复不及时,可发邮件给我: tiehan@sina.cn

Sam avatar
About Sam
专注生物信息 专注转化医学