zoukankan      html  css  js  c++  java
  • tensorflow会话控制-【老鱼学tensorflow】

    在tensorflow中,当定义好结构后,就要通过tf.session()来建立运行时的会话。
    本例子应该不难理解,我们用tensorflow来计算一下一个1行2列的矩阵和2行1列矩阵的乘积:

    import tensorflow as tf
    
    # 1行2列的矩阵
    mat1 = tf.constant([[3, 3]])
    # 2行1列的矩阵
    mat2 = tf.constant([[2],
                        [2]])
    
    out = tf.matmul(mat1, mat2)
    
    sess = tf.Session()
    print(sess.run(out))
    sess.close()
    

    输出为:

    [[12]]
    

    当然,如果读者熟悉python中的with语句,则也可以用with语句来进行,这样就可以不用管sess.close()方法了,因为with会考虑到自动调用相应资源的释放函数

    with tf.Session() as sess:
        print(sess.run(out))
    

    这里顺便也用numpy的方式来计算一下相同的矩阵乘法:

    import numpy as np
    mat1 = np.array([[3, 3]])
    mat2 = np.array([[2],
                     [2]])
    print(np.dot(mat1, mat2))
    

    输出也一样为:

    [[12]]
    
  • 相关阅读:
    第03组 Alpha冲刺(3/6)
    第03组 Alpha冲刺(2/6)
    第03组 Alpha冲刺(1/6)
    团队项目-选题报告
    第3组 团队展示
    福大软工 · BETA 版冲刺前准备(团队)
    Alpha 事后诸葛亮
    Alpha冲刺
    Alpha冲刺-(9/10)
    Alpha冲刺
  • 原文地址:https://www.cnblogs.com/dreampursuer/p/8011589.html
Copyright © 2011-2022 走看看