No need for element wise