Add special case for *Vector in mat64.Norm

gonum · Oct 7, 2015 · b75cdca · b75cdca
1 parent 5fe328e
commit b75cdca
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/mat64/matrix.go b/mat64/matrix.go
@@ -694,6 +694,26 @@ func Norm(a Matrix, norm float64) float64 {
 			work = make([]float64, rm.N)
 		}
 		return lapack64.Lansy(n, rm, work)
+	case *Vector:
+		rv := rma.RawVector()
+		switch norm {
+		default:
+			panic("unreachable")
+		case 1:
+			if aTrans {
+				imax := blas64.Iamax(rma.n, rv)
+				return math.Abs(rma.At(imax, 0))
+			}
+			return blas64.Asum(rma.n, rv)
+		case 2:
+			return blas64.Nrm2(rma.n, rv)
+		case math.Inf(1):
+			if aTrans {
+				return blas64.Asum(rma.n, rv)
+			}
+			imax := blas64.Iamax(rma.n, rv)
+			return math.Abs(rma.At(imax, 0))
+		}
 	}
 	switch norm {
 	default: