| 1234567891011121314151617 |
- // Package ops provides basic tensor operations for CPU backend
- package ops
- import (
- "makarna/pkg/backend/cpu"
- )
- // Add performs element-wise addition: dst += src
- func Add(dst, src *cpu.Tensor) error {
- dstData := dst.DataFloat32()
- srcData := src.DataFloat32()
-
- for i := range dstData {
- dstData[i] += srcData[i]
- }
- return nil
- }
|