一个实例(矩阵乘法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