Linear activation function pytorch.