-
Notifications
You must be signed in to change notification settings - Fork 20
/
vec_parameter.go
67 lines (51 loc) · 1.4 KB
/
vec_parameter.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
62
63
64
65
66
67
package op
// We need to implement a slice based parameter
type sizeIndexIterator struct {
idx, size int
}
func (s *sizeIndexIterator) Index() int {
return s.idx
}
func (s *sizeIndexIterator) Next() bool {
s.idx += 1
return s.idx < s.size
}
func (s *sizeIndexIterator) Rewind() {
s.idx = -1
}
func (s *sizeIndexIterator) Size() int {
return s.size
}
func MakeRangeIndexIterator(psize int) IndexIterator {
return &sizeIndexIterator{idx: -1, size: psize}
}
type sliceParameter struct {
param []float32
}
func (s *sliceParameter) Get(index int) float32 {
return s.param[index]
}
func (s *sliceParameter) Set(index int, value float32) {
s.param[index] = value
}
func (s *sliceParameter) Add(index int, value float32) {
s.param[index] += value
}
// This allow us to generate parameter with same width.
func (s *sliceParameter) CloneWithoutCopy() Parameter {
return &sliceParameter{param: make([]float32, len(s.param), len(s.param))}
}
// This allow one to enumerate through all parameters
func (s *sliceParameter) IndexIterator() IndexIterator {
return &sizeIndexIterator{idx: -1, size: len(s.param)}
}
// This creates a new Vector based parameter
func NewVecParameter(size int) Parameter {
return &sliceParameter{param: make([]float32, size, size)}
}
func NewVecParameterWithData(data []float32) Parameter {
return &sliceParameter{param: data}
}
func (s *sliceParameter) Data() []float32 {
return s.param
}