tensorflow教程之tf.matmul(a,b)的用法，一次性植入你的大脑

tian_jiangnan

于 2020-06-06 11:21:35 发布

阅读量4.3k

点赞数 9

CC 4.0 BY-SA版权

分类专栏： python编程专栏

本文链接：https://blog.csdn.net/tian_jiangnan/article/details/106584921

python编程专栏专栏收录该内容

66 篇文章

订阅专栏

本文详细解释了TensorFlow中tf.matmul函数的使用方法，通过具体示例展示了矩阵相乘的操作过程，并对比了tf.multiply函数的功能区别，强调了矩阵维度匹配的重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

tf.matmul(a,b)是一种矩阵相乘，这个涉及到了线性代数里的知识

举个例子

import tensorflow as tf

a = tf.constant([1, 2, 3, 4, 5, 6], shape=[2, 3])
b = tf.constant([7, 8, 9, 10, 11, 12], shape=[3, 2])
d=tf.matmul(a,b)
with tf.Session() as sess:
    print(sess.run(d))

这个怎么计算呢？

我们现在运行一下程序看看，是否与我们说的一样

一样的！

但是我们需要注意

tf.matmul(a,b)与tf.multiply(a,b)的区别，后者是单纯的对应位置数字相乘

报错的原因是a与b的维度不一样，必须维度一样才行