makarna/pkg/backend/cpu/nn/tensor_utils.go

package nn

import (
	"fmt"

	"makarna/pkg/backend/cpu"
)

func FlattenVector(t *cpu.Tensor, n int, name string) ([]float32, error) {
	if t == nil {
		return nil, fmt.Errorf("missing %s", name)
	}
	data := t.DataFloat32()
	if len(data) == n {
		return data, nil
	}
	if len(data) >= n {
		return data[:n], nil
	}
	return nil, fmt.Errorf("%s has unexpected size %d", name, len(data))
}