1
0

add.go 337 B

1234567891011121314151617
  1. // Package ops provides basic tensor operations for CPU backend
  2. package ops
  3. import (
  4. "makarna/pkg/backend/cpu"
  5. )
  6. // Add performs element-wise addition: dst += src
  7. func Add(dst, src *cpu.Tensor) error {
  8. dstData := dst.DataFloat32()
  9. srcData := src.DataFloat32()
  10. for i := range dstData {
  11. dstData[i] += srcData[i]
  12. }
  13. return nil
  14. }