Add sigmoid, create_constant_array. Fix neg. Fix minor issues

Author: roaffix

arrayfire_wrapper/lib/__init__.py

@@ -41,13 +41,14 @@
 
 from .create_and_modify_array.assignment_and_indexing.lookup import lookup
 
-__all__ += [
-    "constant",
-    "constant_complex",
-    "constant_long",
-    "constant_ulong",
-]
-from .create_and_modify_array.create_array.constant import constant, constant_complex, constant_long, constant_ulong
+__all__ += ["constant", "constant_complex", "constant_long", "constant_ulong", "create_constant_array"]
+from .create_and_modify_array.create_array.constant import (
+    constant,
+    constant_complex,
+    constant_long,
+    constant_ulong,
+    create_constant_array,
+)
 
 __all__ += ["diag_create", "diag_extract"]
 
@@ -539,6 +540,7 @@
     "root",
     "rsqrt",
     "sqrt",
+    "sigmoid",
     "tgamma",
 ]
 
@@ -558,6 +560,7 @@
     pow2,
     root,
     rsqrt,
+    sigmoid,
     sqrt,
     tgamma,
 )

arrayfire_wrapper/lib/create_and_modify_array/create_array/__init__.py

@@ -1,7 +1,7 @@
 # flake8: noqa
-__all__ = ["constant", "constant_complex", "constant_long", "constant_ulong"]
+__all__ = ["constant", "constant_complex", "constant_long", "constant_ulong", "create_constant_array"]
 
-from .constant import constant, constant_complex, constant_long, constant_ulong
+from .constant import constant, constant_complex, constant_long, constant_ulong, create_constant_array
 
 __all__ += [
     "create_random_engine",

arrayfire_wrapper/lib/create_and_modify_array/create_array/constant.py

@@ -1,7 +1,7 @@
 import ctypes
 
 from arrayfire_wrapper.defines import AFArray, CShape
-from arrayfire_wrapper.dtypes import Dtype
+from arrayfire_wrapper.dtypes import Dtype, complex32, complex64, implicit_dtype, int64, uint64
 from arrayfire_wrapper.lib._utility import call_from_clib
 
 
@@ -76,3 +76,19 @@ def constant_ulong(number: int | float, shape: tuple[int, ...], dtype: Dtype, /)
         ctypes.pointer(c_shape.c_array),
     )
     return out
+
+
+def create_constant_array(number: int | float | complex, shape: tuple[int, ...], dtype: Dtype, /) -> AFArray:
+    if not dtype:
+        dtype = implicit_dtype(number, dtype)
+
+    if isinstance(number, complex):
+        return constant_complex(number, shape, dtype if dtype in {complex32, complex64} else complex32)
+
+    if dtype == int64:
+        return constant_long(number, shape, dtype)
+
+    if dtype == uint64:
+        return constant_ulong(number, shape, dtype)
+
+    return constant(number, shape, dtype)

arrayfire_wrapper/lib/mathematical_functions/__init__.py

@@ -25,6 +25,7 @@
     pow2,
     root,
     rsqrt,
+    sigmoid,
     sqrt,
     tgamma,
 )
@@ -46,6 +47,7 @@
     "root",
     "rsqrt",
     "sqrt",
+    "sigmoid",
     "tgamma",
 ]
 

arrayfire_wrapper/lib/mathematical_functions/exp_and_log_functions.py

@@ -114,6 +114,13 @@ def sqrt(arr: AFArray, /) -> AFArray:
     return unary_op(sqrt.__name__, arr)
 
 
+def sigmoid(arr: AFArray, /) -> AFArray:
+    """
+    source: https://arrayfire.org/docs/group__arith__func__sigmoid.htm#gadf4280e3283b65264de75194e0a6d565
+    """
+    return unary_op(sigmoid.__name__, arr)
+
+
 def tgamma(arr: AFArray, /) -> AFArray:
     """
     source:

arrayfire_wrapper/lib/mathematical_functions/numeric_functions.py

@@ -1,7 +1,9 @@
 import ctypes
 
 from arrayfire_wrapper.defines import AFArray
+from arrayfire_wrapper.dtypes import float32
 from arrayfire_wrapper.lib._utility import binary_op, call_from_clib, unary_op
+from arrayfire_wrapper.lib.create_and_modify_array.create_array import create_constant_array
 from arrayfire_wrapper.lib.mathematical_functions.arithmetic_operations import sub
 
 
@@ -46,7 +48,9 @@ def hypot(lhs: AFArray, rhs: AFArray, /) -> AFArray:
     """
     source:
     """
-    return binary_op(hypot.__name__, lhs, rhs)
+    out = AFArray.create_null_pointer()
+    call_from_clib(hypot.__name__, lhs, rhs)
+    return out
 
 
 def maxof(lhs: AFArray, rhs: AFArray, /) -> AFArray:
@@ -71,7 +75,7 @@ def mod(lhs: AFArray, rhs: AFArray, /) -> AFArray:
 
 
 def neg(arr: AFArray) -> AFArray:
-    return sub(AFArray(0), arr)
+    return sub(create_constant_array(0, (1,), float32), arr)
 
 
 def rem(lhs: AFArray, rhs: AFArray, /) -> AFArray:

arrayfire_wrapper/version.py

@@ -1,7 +1,7 @@
 import os
 
 _MAJOR = "0"
-_MINOR = "4"
+_MINOR = "5"
 # On main and in a nightly release the patch should be one ahead of the last
 # released build.
 _PATCH = "0"