forked from arrayfire/arrayfire
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathstatistics.cpp
More file actions
103 lines (86 loc) · 3.2 KB
/
statistics.cpp
File metadata and controls
103 lines (86 loc) · 3.2 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
/*******************************************************
* Copyright (c) 2015, ArrayFire
* All rights reserved.
*
* This file is distributed under 3-clause BSD license.
* The complete license agreement can be obtained at:
* http://arrayfire.com/licenses/BSD-3-Clause
********************************************************/
#include <af/array.h>
#include <af/statistics.h>
#include "symbol_manager.hpp"
af_err af_mean(af_array *out, const af_array in, const dim_t dim) {
CHECK_ARRAYS(in);
CALL(af_mean, out, in, dim);
}
af_err af_mean_weighted(af_array *out, const af_array in,
const af_array weights, const dim_t dim) {
CHECK_ARRAYS(in, weights);
CALL(af_mean_weighted, out, in, weights, dim);
}
af_err af_var(af_array *out, const af_array in, const bool isbiased,
const dim_t dim) {
CHECK_ARRAYS(in);
CALL(af_var, out, in, isbiased, dim);
}
af_err af_var_weighted(af_array *out, const af_array in, const af_array weights,
const dim_t dim) {
CHECK_ARRAYS(in, weights);
CALL(af_var_weighted, out, in, weights, dim);
}
af_err af_meanvar(af_array *mean, af_array *var, const af_array in,
const af_array weights, const af_var_bias bias,
const dim_t dim) {
CHECK_ARRAYS(in, weights);
CALL(af_meanvar, mean, var, in, weights, bias, dim);
}
af_err af_stdev(af_array *out, const af_array in, const dim_t dim) {
CHECK_ARRAYS(in);
CALL(af_stdev, out, in, dim);
}
af_err af_cov(af_array *out, const af_array X, const af_array Y,
const bool isbiased) {
CHECK_ARRAYS(X, Y);
CALL(af_cov, out, X, Y, isbiased);
}
af_err af_median(af_array *out, const af_array in, const dim_t dim) {
CHECK_ARRAYS(in);
CALL(af_median, out, in, dim);
}
af_err af_mean_all(double *real, double *imag, const af_array in) {
CHECK_ARRAYS(in);
CALL(af_mean_all, real, imag, in);
}
af_err af_mean_all_weighted(double *real, double *imag, const af_array in,
const af_array weights) {
CHECK_ARRAYS(in, weights);
CALL(af_mean_all_weighted, real, imag, in, weights);
}
af_err af_var_all(double *realVal, double *imagVal, const af_array in,
const bool isbiased) {
CHECK_ARRAYS(in);
CALL(af_var_all, realVal, imagVal, in, isbiased);
}
af_err af_var_all_weighted(double *realVal, double *imagVal, const af_array in,
const af_array weights) {
CHECK_ARRAYS(in, weights);
CALL(af_var_all_weighted, realVal, imagVal, in, weights);
}
af_err af_stdev_all(double *real, double *imag, const af_array in) {
CHECK_ARRAYS(in);
CALL(af_stdev_all, real, imag, in);
}
af_err af_median_all(double *realVal, double *imagVal, const af_array in) {
CHECK_ARRAYS(in);
CALL(af_median_all, realVal, imagVal, in);
}
af_err af_corrcoef(double *realVal, double *imagVal, const af_array X,
const af_array Y) {
CHECK_ARRAYS(X, Y);
CALL(af_corrcoef, realVal, imagVal, X, Y);
}
af_err af_topk(af_array *values, af_array *indices, const af_array in,
const int k, const int dim, const af_topk_function order) {
CHECK_ARRAYS(in);
CALL(af_topk, values, indices, in, k, dim, order);
}