牧者

大风起兮云飞扬

  博客园 :: 首页 :: 新随笔 :: 联系 :: 订阅 :: 管理 ::

TF 常用函数说明01

tf.nn.relu(features, name = None)

这个函数的作用是计算激活函数 relu,即 max(features, 0)。将大于0的保持不变,小于0的数置为0。

import tensorflow as tf

a = tf.constant([-1.0, 2.0])
with tf.Session() as sess:
    b = tf.nn.relu(a)
    print sess.run(b)
以上程序输出的结果是:[0. 2.]

tf.multiply()两个矩阵中对应元素各自相乘

tf.matmul()将矩阵a乘以矩阵b,生成a * b。

 

tf.assign()  赋值

tf.cast()数据类型转换

 

 

 

 

 

 

  

posted on 2019-12-12 11:46  牧者.D  阅读(242)  评论(0)    收藏  举报