Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-05 17:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add utils vector mean function

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-14 21:41:18 +02:00
parent 5c8d3f7e25
commit da8f3167cb
4 changed files with 71 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
include/utils/vector.cuh
View File

@@ -38,6 +38,17 @@ void sum(const float *d_vec, float *d_sum, const unsigned int length);
*/
void max(const float *d_vec, float *d_max, const unsigned int length);
/**
* @brief Compute the mean of the vector
*
* @param d_vec Device pointer to the vector
* @param d_mean Device pointer to the mean
* @param d_length Device pointer to the length
* @param length Length of the vector
*/
void mean(const float *d_vec, float *d_mean, float *d_length, int length);
// /**
// * @brief Compute the variance of a vector
// *