- package ops
- import "makarna/pkg/backend/cpu"
- // Mul performs element-wise multiplication: dst *= src
- func Mul(dst, src *cpu.Tensor) error {
- dstData := dst.DataFloat32()
- srcData := src.DataFloat32()
-
- for i := range dstData {
- dstData[i] *= srcData[i]
- }
- return nil
- }
|