tensorflow--tf实现矩阵乘法和加法

tf.add()和tf.matmul()的简单实现

import tensorflow as tf

# tf2默认为即时执行模式

# TensorFlow 使用张量(tensor)作为数据的基本单位。TensorFlow 的张量在概念上等同于多维数组,
# 我们可以使用它来描述数学中的标量(零维数组)、向量(一维数组)、矩阵(二维数组)等
# 定义一个随机数(标量)
random_float = tf.random.uniform(shape=())
print(random_float)
random_int = tf.random.uniform(maxval=10, dtype=tf.int32, shape=())
print(random_int)

# 定义一个有2个元素的零向量
zero_vector = tf.zeros(shape=(2))

# 定义两个2×2的常量矩阵
A = tf.constant([[1., 2.], [3., 4.]])
B = tf.constant([[5., 6.], [7., 8.]])

# 张量的重要属性是形状、类型和值,它们分别可以通过张量的 shape、dtype 属性和numpy()方法获得。
# 查看矩阵A的形状、类型和值
print(A.shape)      # 输出(2, 2),即矩阵的长和宽均为2
print(A.dtype)      # 输出<dtype: 'float32'>
# 张量的numpy()方法是将张量的值转换为一个 NumPy数组。
print(A.numpy())    # 输出[[1. 2.]
                    #      [3. 4.]]

C = tf.add(A, B)    # 计算矩阵A和B的和
D = tf.matmul(A, B) # 计算矩阵A和B的乘积

print(C)
print(D)

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值