INNER CODE UNIT · Go

ScaleVector

takara-ai/go-attention · attention/attention.go:118

func ScaleVector(v Vector, scale float64) Vector {
	result := make(Vector, len(v))
	for i, val := range v {
		result[i] = val * scale
	}
	return result
}

// AddVectors adds two vectors element-wise
func AddVectors(v1, v2 Vector) (Vector, error) {
	if len(v1) != len(v2) {
		return nil, fmt.Errorf("vector dimensions mismatch: %d != %d", len(v1), len(v2))
	}

	result := make(Vector, len(v1))
	for i := range v1 {
		result[i] = v1[i] + v2[i]
	}

View source record →