-
Notifications
You must be signed in to change notification settings - Fork 0
/
fp32.go
61 lines (50 loc) · 1.03 KB
/
fp32.go
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
package gotensor
import "github.com/chewxy/math32"
func (*GoTensor) FP32Dot(x, w []float32) float32 {
return dotVector(x, w, int64(len(x)))
}
func (*GoTensor) FP32Mul(x, w, y []float32) {
for i := range x {
y[i] = x[i] * w[i]
}
}
func (*GoTensor) FP32ScalarMul(x float32, w, y []float32) {
for i := range w {
y[i] = x * w[i]
}
}
func (*GoTensor) FP32Div(x, w, y []float32) {
for i := range x {
y[i] = x[i] / w[i]
}
}
func (*GoTensor) FP32ScalarDiv(x float32, w, y []float32) {
for i := range w {
y[i] = x / w[i]
}
}
func (*GoTensor) FP32Add(x, w, y []float32) {
for i := range x {
y[i] = x[i] + w[i]
}
}
func (*GoTensor) FP32ScalarAdd(x float32, w, y []float32) {
for i := range w {
y[i] = x + w[i]
}
}
func (*GoTensor) FP32Sub(x, w, y []float32) {
for i := range x {
y[i] = x[i] - w[i]
}
}
func (*GoTensor) FP32ScalarSub(x float32, w, y []float32) {
for i := range w {
y[i] = x - w[i]
}
}
func (*GoTensor) FP32Pow(x []float32, n float32, y []float32) {
for i := range x {
y[i] = math32.Pow(x[i], n)
}
}