TensorFlow 基础练习 1 矩阵点乘

最新推荐文章于 2025-04-17 15:37:49 发布

原创最新推荐文章于 2025-04-17 15:37:49 发布 · 920 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #python

python 编程专栏收录该内容

8 篇文章

订阅专栏

本文详细介绍使用TensorFlow进行矩阵点乘的操作方法。通过具体示例，演示如何定义矩阵并利用tf.matmul函数实现两个矩阵的点乘运算。适用于初学者理解和掌握TensorFlow的基本矩阵运算。

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

1、矩阵点乘

# 计算一个矩阵相乘
import tensorflow as tf

m1 = tf.constant([[1, 2]], dtype=tf.int32)  # 定义常量张量大小为 1x2, 类型为 tf.int32
m2 = tf.constant([[1],
                  [2]])                    # 定义常量张量大小为 2x1， 类型 tf.int32

dot_operation = tf.matmul(m1, m2)          # tf.matmul函数完成两个张量的点乘操作

with tf.Session() as sess:
    result = sess.run(dot_operation)       # 执行会话，得到运算的结果
    print(result)

结果：

[[5]]