new data generation function: constant_like

function to create an array with same shape and type as input array filled with a constant

Author: 9prady9

src/data/mod.rs

@@ -622,3 +622,26 @@ pub fn replace_scalar(a: &mut Array, cond: &Array, b: f64) {
         HANDLE_ERROR(AfError::from(err_val));
     }
 }
+
+/// Create an array filled with given constant retaining type/shape of another Array.
+///
+/// # Parameters
+///
+/// - `value` is the constant with which output Array is to be filled
+/// - `input` is the Array whose shape the output Array has to maintain
+///
+/// # Return Values
+///
+/// Array with given constant value and input Array's shape and similar internal data type.
+pub fn constant_like(value: f64, input: &Array) -> Array {
+    let dims = input.dims();
+    unsafe {
+        let mut temp: i64 = 0;
+        let err_val = af_constant(&mut temp as MutAfArray, value as c_double,
+                                  dims.ndims() as c_uint,
+                                  dims.get().as_ptr() as *const DimT,
+                                  input.get_type() as c_int);
+        HANDLE_ERROR(AfError::from(err_val));
+        Array::from(temp)
+    }
+}

src/lib.rs

@@ -32,7 +32,7 @@ mod backend;
 pub use blas::{matmul, dot, transpose, transpose_inplace};
 mod blas;
 
-pub use data::{constant, range, iota};
+pub use data::{constant, constant_like, range, iota};
 pub use data::{identity, diag_create, diag_extract, lower, upper};
 pub use data::{join, join_many, tile};
 pub use data::{reorder, shift, moddims, flat, flip};