TensorFlow矩阵乘法-认识Session

版权声明:本文为博主原创文章,欢迎转载。 https://blog.csdn.net/samylee/article/details/84617835

tensorflow教程顺序有点乱,其实这个应该放在第一篇的,大伙凑合看吧^_^

硬件:NVIDIA-GTX1080

软件:Windows7、python3.6.5、tensorflow-gpu-1.4.0

一、基础知识

矩阵乘法规则:1x2的matrix和2x1的matrix相乘,得到1x1的matrix

二、代码展示

import tensorflow as tf

matrix1 = tf.constant([[3,3]])
matrix2 = tf.constant([[2],
                       [2]])
product = tf.matmul(matrix1, matrix2)

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

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

三、运行结果

[[12]]

任何问题请加唯一QQ2258205918(名称samylee)!

猜你喜欢

转载自blog.csdn.net/samylee/article/details/84617835