def get_activation_function(activation_function):
if not activation_function:
return lambda a: a
try:
return getattr(tf.nn, activation_function)
except AttributeError:
raise ValueError(
'Invalid activation function "{}"'.format(activation_function))
评论列表
文章目录