add axis check for elementwise op while the dimension of x is equal to the dimension of tensor (#35340)