Add square and cube operators

docs/operators.md

@@ -14,6 +14,8 @@ of these and other valid operators are stated below.
 **Unary**
 
 `neg`,
+`square`,
+`cube`,
 `exp`,
 `abs`,
 `logm` (=log(abs(x) + 1e-8)),

julia/operators.jl

@@ -16,6 +16,8 @@ import Base.FastMath: sqrt_llvm_fast, neg_float_fast,
 plus(x::Float32, y::Float32)::Float32 = add_float_fast(x, y) #Do not change the name of this operator.
 sub(x::Float32, y::Float32)::Float32 = sub_float_fast(x, y) #Do not change the name of this operator.
 mult(x::Float32, y::Float32)::Float32 = mul_float_fast(x, y) #Do not change the name of this operator.
+square(x::Float32)::Float32 = mul_float_fast(x, x)
+cube(x::Float32)::Float32 = mul_float_fast(mul_float_fast(x, x), x)
 pow(x::Float32, y::Float32)::Float32 = sign_fast(x)*pow_fast(abs(x), y)
 div(x::Float32, y::Float32)::Float32 = div_float_fast(x, y)
 logm(x::Float32)::Float32 = log_fast(abs_fast(x) + 1f-8)

pysr/sr.py

@@ -16,6 +16,8 @@ global_extra_sympy_mappings = {}
 sympy_mappings = {
     'div':  lambda x, y : x/y,
     'mult': lambda x, y : x*y,
+    'square':lambda x   : x**2,
+    'cube': lambda x    : x**3,
     'plus': lambda x, y : x + y,
     'sub':  lambda x, y : x - y,
     'neg':  lambda x    : -x,