如何在tf.layers.dense中使用leaky_relu

  • 因爲leaky_relu不支持字符形式,即
# this is wriong
tf.layers.Dense(units,activation='leaky_relu')
# right operation
import tensorflow as tf
from functools import partial

output = tf.layers.dense(input, n_units, activation=partial(tf.nn.leaky_relu, alpha=0.01))
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章