pub fn square() -> ElementWiseOp