def mlp_func(self):
func = self._config.get('Functions', 'mlp_func')
if func == 'identity':
return tf.identity
else:
return getattr(tf.nn, func)
微信
公众号
扫码关注公众号