def simple_soft_threshold(r_, lam_):
"implement a soft threshold function y=sign(r)*max(0,abs(r)-lam)"
lam_ = tf.maximum(lam_, 0)
return tf.sign(r_) * tf.maximum(tf.abs(r_) - lam_, 0)
微信
公众号
扫码关注公众号