| 33 | } |
| 34 | |
| 35 | static array dist_vec(array a, array b) { |
| 36 | array dist_mat = constant(0, (int)a.dims(1), (int)b.dims(1)); |
| 37 | |
| 38 | // Iterate through columns a |
| 39 | for (int ii = 0; ii < (int)a.dims(1); ii++) { |
| 40 | array avec = a(span, ii); |
| 41 | |
| 42 | // Iterate through columns of b |
| 43 | for (int jj = 0; jj < (int)b.dims(1); jj++) { |
| 44 | array bvec = b(span, jj); |
| 45 | |
| 46 | // get SAD using sum on the vector |
| 47 | dist_mat(ii, jj) = sum(abs(avec - bvec)); |
| 48 | } |
| 49 | } |
| 50 | |
| 51 | return dist_mat; |
| 52 | } |
| 53 | |
| 54 | static array dist_gfor1(array a, array b) { |
| 55 | array dist_mat = constant(0, (int)a.dims(1), (int)b.dims(1)); |