tensor/dense_reduction_methods.go at master · pdevine/tensor · GitHub

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
package tensor

import "github.com/pkg/errors"

func (t *Dense) Sum(along ...int) (retVal *Dense, err error) {
	var e Engine = t.e
	if sumer, ok := e.(Sumer); ok {
		var ret Tensor
		if ret, err = sumer.Sum(t, along...); err != nil {
			return
		}
		return ret.(*Dense), nil
	}
	return nil, errors.Errorf("Engine does not support Sum")
}

func (t *Dense) Max(along ...int) (retVal *Dense, err error) {
	var e Engine = t.e
	if maxer, ok := e.(Maxer); ok {
		var ret Tensor
		if ret, err = maxer.Max(t, along...); err != nil {
			return
		}
		return ret.(*Dense), nil
	}
	return nil, errors.Errorf("Engine does not support Max")
}

func (t *Dense) Min(along ...int) (retVal *Dense, err error) {
	var e Engine = t.e
	if miner, ok := e.(Miner); ok {
		var ret Tensor
		if ret, err = miner.Min(t, along...); err != nil {
			return
		}
		return ret.(*Dense), nil
	}
	return nil, errors.Errorf("Engine does not support Min")
}