tensorflow2.1下的tf.nn.relu()函数

tf.nn.relu()函数是将大于0的数保持不变,小于0的数置为0:
f(x)=max(0,x)f(x) = max(0, x)

CSDN图标
relu
(
    features,
    name=None
)

参数说明:

features - tensor类型,类型为float32, float64, int32, int64, uint8, int16, int8, uint16, half

name - 可选,string类型,运算名称

示例代码:

import  os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
import tensorflow as tf

a = tf.constant([1, -2, 0, 4, -5, 6])
b = tf.nn.relu(a)
print(b)
# tf.Tensor([1 0 0 4 0 6], shape=(6,), dtype=int32)
发布了79 篇原创文章 · 获赞 53 · 访问量 8352
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 技术黑板 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览