tensorflow矩阵的运算

一个实例(矩阵乘法matmul)

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' #CPU的分配问题
import numpy
"""
Please note, this code is only for python 3+. If you are using python 2+, please modify the code accordingly.
"""
import tensorflow as tf

matrix1 = tf.constant([[3, 3]])
matrix2 = tf.constant([[2],
                       [2]])
product = tf.matmul(matrix1, matrix2)  # matrix multiply np.dot(m1, m2)

# method 1
sess = tf.compat.v1.Session()
result = sess.run(product)
print(result)
sess.close()

# method 2
with tf.compat.v1.Session() as sess:
    result2 = sess.run(product)
    print(result2)

矩阵乘法(multiply)

#初始化cup的分配
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

"""
Please note, this code is only for python 3+. If you are using python 2+, please modify the code accordingly.
"""
import tensorflow as tf

input1 = tf.compat.v1.placeholder(tf.float32)
input2 = tf.compat.v1.placeholder(tf.float32)
ouput = tf.multiply(input1, input2)

with tf.compat.v1.Session() as sess:
    print(sess.run(ouput, feed_dict={input1: [7.], input2: [2.]}))

matmul和multiply区别参考文章:https://blog.csdn.net/qq_29678299/article/details/89367600

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值