diff --git a/matrix.go b/matrix.go index 34466c5..3f7714f 100644 --- a/matrix.go +++ b/matrix.go @@ -164,6 +164,14 @@ func (m *Matrix) Add(other *Matrix) (*Matrix, error) { return NewMatrix(resultData, m.shape) } +func (m *Matrix) Sum() float64 { + var sum float64 = 0 + for i := 0; i < m.size; i++ { + sum += m.data[i] + } + return sum +} + // Subtract 矩阵减法 // other: 要相减的另一个矩阵 func (m *Matrix) Subtract(other *Matrix) (*Matrix, error) {