Add square fast path

gonum · Dec 15, 2014 · 189d7a2 · 189d7a2
1 parent 73d2610
commit 189d7a2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/mat64/dense_arithmetic.go b/mat64/dense_arithmetic.go
@@ -474,6 +474,9 @@ func (m *Dense) Pow(a Matrix, n int) {
 	case 1:
 		m.Copy(a)
 		return
+	case 2:
+		m.Mul(a, a)
+		return
 	}
 
 	// Perform iterative exponentiation by squaring in work space.