How does one make a quadratic activation function?

1

is

a = x.mm(W)**2

really all one needs? Its so simply that I just want to double check it.