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

  • 因为leaky_relu不支持字符形式,即
# this is wrong
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))

你可能感兴趣的:(tensorflow学习)