def activation_from_string(activation_str):
if activation_str is None:
return tf.identity
return getattr(tf.nn, activation_str)
微信
公众号
扫码关注公众号