Polynomial Tensor Sketch for Element-wise Matrix Function