Rewrite feature selected tanh nn