Pytorch tanh function