Implement vector mean calculation

include/kernels/matmul.cuh

@@ -37,6 +37,29 @@ __global__ void vec_vec_add(
     const unsigned int w
 );
 
+/**
+ * @brief Vector vector subtraction kernel
+ * 
+ * @param d_vector1 
+ * @param d_vector2 
+ * @param d_output 
+ * @param w 
+ * @return __global__ 
+ */
+__global__ void vec_vec_sub(
+    const float* __restrict__ d_vector1,
+    const float* __restrict__ d_vector2,
+    float* __restrict__ d_output,
+    const unsigned int w
+);
+
+__global__ void vec_vec_mul(
+    const float* __restrict__ d_vector1,
+    const float* __restrict__ d_vector2,
+    float* __restrict__ d_output,
+    const unsigned int w
+);
+
 /**
  * @brief Sub scalar from each element of the vector
  * 

include/utils/vector.cuh

@@ -31,13 +31,30 @@ void sum(float *d_vec, float *d_sum, const unsigned int length);
 
 
 /**
- * @brief Utility function that returns the max of a vector
+ * @brief Get the max of a vector
  * 
  * @param d_vec Pointer to the vector
  * @param length Length of the vector
  */
 void max(float *d_vec, float *d_max, const unsigned int length);
 
+/**
+ * @brief Compute the mean of a vector
+ * 
+ * @param d_vec 
+ * @param d_mean 
+ * @param length 
+ */
+void mean(float *d_vec, float *d_mean, const unsigned int length);
+
+/**
+ * @brief Compute the variance of a vector
+ * 
+ * @param d_vec 
+ * @param d_var 
+ * @param length 
+ */
+void var(float *d_vec, float *d_var, const unsigned int length);
 
 }  // namespace CUDANet::Utils