use float32

This commit is contained in:
Roy Han
2024-07-02 10:30:29 -07:00
parent 512e0a7bde
commit 00a4cb26ca
6 changed files with 27 additions and 27 deletions

View File

@@ -2,18 +2,18 @@ package format
import "math"
func Normalize(vec []float64) []float64 {
func Normalize(vec []float32) []float32 {
var sum float64
for _, v := range vec {
sum += v * v
sum += float64(v * v)
}
sum = math.Sqrt(sum)
var norm float64
var norm float32
if sum > 0 {
norm = 1.0 / sum
norm = float32(1.0 / sum)
} else {
norm = 0.0
}