糖尿病康复,内容丰富有趣,生活中的好帮手!
糖尿病康复 > tensorflow2.0使用GELU激活函数

tensorflow2.0使用GELU激活函数

时间:2021-02-03 16:57:33

相关推荐

tensorflow2.0使用GELU激活函数

首先,tensorflow2.0没有GELU激活函数,所以必须自定义GELU激活函数。

其次,自定义代码如下:

第一步:import tensorflow as tf

第二步:class GELU(layers.Layer):def __init__(self):super(GELU, self).__init__()def call(self, x):cdf = 0.5 * (1.0 + tf.tanh((np.sqrt(2 / np.pi) * (x + 0.044715 * tf.pow(x, 3)))))return x * cdf

第三步:

self.act=GRLU()

第四步:(call中使用)

x = self.act(x)

如果觉得《tensorflow2.0使用GELU激活函数》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。