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

你可能感兴趣的:(TensorFlow)