FEAT: Adding .as_type method to Array class

Author: pavanky

arrayfire/array.py

@@ -423,6 +423,15 @@ def __init__(self, src=None, dims=(0,), dtype=None, is_device=False):
                 idims[n] = dims[n]
             self.arr = _create_empty_array(numdims, idims, to_dtype[type_char])
 
+    def as_type(self, ty):
+        """
+        Cast current array to a specified data type
+
+        Parameters
+        ----------
+        ty : Return data type
+        """
+        return cast(self, ty)
 
     def copy(self):
         """
@@ -1134,3 +1143,4 @@ def read_array(filename, index=None, key=None):
     return out
 
 from .algorithm import sum
+from .arith import cast

tests/simple/array_test.py

@@ -18,6 +18,10 @@ def simple_array(verbose=False):
 
     a = af.Array([1, 2, 3])
     display_func(a)
+
+    b = a.as_type(af.Dtype.s32)
+    display_func(b)
+
     print_func(a.elements(), a.type(), a.dims(), a.numdims())
     print_func(a.is_empty(), a.is_scalar(), a.is_column(), a.is_row())
     print_func(a.is_complex(), a.is_real(), a.is_double(), a.is_single())