def norm(x):
return x * tf.rsqrt(tf.reduce_mean(tf.square(x), keep_dims=True))
微信
公众号
扫码关注公众号
微信
公众号