mul.go 274 B

1234567891011121314
  1. package ops
  2. import "makarna/pkg/backend/cpu"
  3. // Mul performs element-wise multiplication: dst *= src
  4. func Mul(dst, src *cpu.Tensor) error {
  5. dstData := dst.DataFloat32()
  6. srcData := src.DataFloat32()
  7. for i := range dstData {
  8. dstData[i] *= srcData[i]
  9. }
  10. return nil
  11. }