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
|
package tensor
import "github.com/pkg/errors"
// Sum sums a Tensor along the given axes
func Sum(t Tensor, along ...int) (retVal Tensor, err error) {
if sumer, ok := t.Engine().(Sumer); ok {
return sumer.Sum(t, along...)
}
return nil, errors.New("Engine does not support Sum()")
}
// Argmax finds the index of the max value along the axis provided
func Argmax(t Tensor, axis int) (retVal Tensor, err error) {
if argmaxer, ok := t.Engine().(Argmaxer); ok {
return argmaxer.Argmax(t, axis)
}
return nil, errors.New("Engine does not support Argmax()")
}
// Argmin finds the index of the min value along the axis provided
func Argmin(t Tensor, axis int) (retVal Tensor, err error) {
if argminer, ok := t.Engine().(Argminer); ok {
return argminer.Argmin(t, axis)
}
return nil, errors.New("Engine does not support Argmax()")
}
|
