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
|
#pragma once
#include "data_math.hpp"
namespace saw {
namespace math {
/*
template<typename T, typename Encoding = encode::Native>
data<typename T::InnerType,Encoding> norm_2(const data<T,Encoding>& d){
return {};
}
*/
template<typename T, uint64_t D, typename Encoding = encode::Native>
data<schema::Scalar<T>, Encoding> dot(const data<schema::Vector<T,D>, Encoding>& left, const data<schema::Vector<T,D>, Encoding>& right){
data<schema::Scalar<T>,Encoding> val;
auto& inner = val({});
for(uint64_t i = 0u; i < D; ++i){
inner = inner + left({{i}}) * right({{i}});
}
return val;
}
template<typename T,typename Encoding = encode::Native>
data<schema::Scalar<T>,Encoding> sqrt(const data<schema::Scalar<T>,Encoding>& inp){
data<schema::Scalar<T>,Encoding> out;
out.at({}).set(std::sqrt(inp.at({}).get()));
return out;
}
template<typename T, uint64_t D, typename Encoding = encode::Native>
data<schema::Vector<T,D>, Encoding> normalize(const data<schema::Vector<T,D>>& input ) {
auto inp_dot = dot<T,D,Encoding>(input,input);
auto sqrt_inp_dot = sqrt<T>(inp_dot);
saw::data<schema::Vector<T,D>, Encoding> out;
out.at({{0u}}).set(input.at({{0u}}).get() / sqrt_inp_dot.at({}).get());
out.at({{1u}}).set(input.at({{1u}}).get() / sqrt_inp_dot.at({}).get());
return out;
}
}
}
|
