tensorflow中自定义激活函数

一、tensorflow中常用的模型定义

import tensorflow as tf

model=tf.keras.Sequential([
    tf.keras.layers.Dense(64, kernel_initializer='normal', activation='relu'),
    tf.keras.layers.Dense(1, kernel_initializer='normal', activation='relu')
)]

上面这段代码定义了两层，输出为一个元素。

如果想实验自己的激活函数，又不想过多地写代码。可以使用以下方法：

比如定义tanh

二、在keras网站上对自定义激活函数有如下说明：

Creating custom activations

You can also use a TensorFlow callable as an activation (in this case it