https://machinelearningmastery.com/activation-functions-in-pytorch/