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
104
105
106
107
108
109
110
111
|
package tensor
import "fmt"
func ExampleSum() {
T := New(WithBacking([]float64{0, 1, 2, 3}), WithShape(2, 2))
fmt.Printf("T:\n%v\n", T)
// sum along axis 0
summed, _ := Sum(T, 0)
fmt.Printf("Summed:\n%v\n", summed)
// to keep dims, simply reshape
summed.Reshape(1, 2)
fmt.Printf("Summed (Kept Dims - Shape: %v):\n%v\n\n", summed.Shape(), summed)
// summing along multiple axes
summed, _ = Sum(T, 1, 0)
fmt.Printf("Summed along (1, 0): %v", summed)
// Output:
// T:
// ⎡0 1⎤
// ⎣2 3⎦
//
// Summed:
// [2 4]
// Summed (Kept Dims - Shape: (1, 2)):
// R[2 4]
//
// Summed along (1, 0): 6
}
func ExampleSum_sliced() {
T := New(WithBacking([]float64{0, 1, 2, 3}), WithShape(2, 2))
fmt.Printf("T:\n%v\n", T)
V, _ := T.Slice(nil, S(1))
fmt.Printf("V:\n%v\n", V)
Σ, _ := Sum(V)
fmt.Printf("Σ: %v", Σ)
// Output:
// T:
// ⎡0 1⎤
// ⎣2 3⎦
//
// V:
// [1 3]
// Σ: 4
}
func ExampleArgmax() {
T := New(WithBacking([]float64{0, 100, 200, 3}), WithShape(2, 2))
fmt.Printf("T:\n%v\n", T)
// argmax along the x-axis
am, _ := Argmax(T, 0)
fmt.Printf("Argmax: %v\n", am)
fmt.Printf("Argmax is %T of %v", am, am.Dtype())
// Output:
// T:
// ⎡ 0 100⎤
// ⎣200 3⎦
//
// Argmax: [1 0]
// Argmax is *tensor.Dense of int
}
func ExampleArgmax_sliced() {
T := New(WithBacking([]float64{0, 100, 200, 3}), WithShape(2, 2))
fmt.Printf("T:\n%v\n", T)
// slice creates a view
V, _ := T.Slice(nil, S(1))
// argmax along the x-axis
am, _ := Argmax(V, 0)
fmt.Printf("Argmax: %v\n", am)
fmt.Printf("Argmax is %T of %v", am, am.Dtype())
// Output:
// T:
// ⎡ 0 100⎤
// ⎣200 3⎦
//
// Argmax: 0
// Argmax is *tensor.Dense of int
}
func ExampleArgmin() {
T := New(WithBacking([]float64{0, 100, 200, 3}), WithShape(2, 2))
fmt.Printf("T:\n%v\n", T)
// argmax along the x-axis
am, _ := Argmin(T, 0)
fmt.Printf("Argmin: %v\n", am)
fmt.Printf("Argmin is %T of %v", am, am.Dtype())
// Output:
// T:
// ⎡ 0 100⎤
// ⎣200 3⎦
//
// Argmin: [0 1]
// Argmin is *tensor.Dense of int
}
|
