def sqrt(x): if isinstance(x, afnumpy.ndarray): s = arrayfire.sqrt(x.d_array) return afnumpy.ndarray(x.shape, dtype=pu.typemap(s.dtype()), af_array=s) else: return numpy.sqrt(x)
def arctan2(x1, x2): if isinstance(x1, afnumpy.ndarray) and isinstance(x2, afnumpy.ndarray): s = arrayfire.atan2(x1.d_array, x2.d_array) return afnumpy.ndarray(x1.shape, dtype=pu.typemap(s.dtype()), af_array=s) else: return numpy.arctan(x1, x2)