Function square_vec 

pub fn square_vec(input: &[f32], result: &mut [f32])

SIMD-optimized element-wise square function

Computes result[i] = input[i] * input[i] for all elements. This is more efficient than using pow_vec with exponent 2.

Arguments

• input - Input vector
• result - Output vector (must have same length as input)

Panics

Panics if input and output vectors have different lengths

Examples

use sklears_simd::vector::math_functions::square_vec;

let input = vec![1.0, 2.0, 3.0, -4.0];
let mut result = vec![0.0; 4];

square_vec(&input, &mut result);
assert!((result[0] - 1.0).abs() < 1e-6);
assert!((result[1] - 4.0).abs() < 1e-6);
assert!((result[2] - 9.0).abs() < 1e-6);
assert!((result[3] - 16.0).abs() < 1e-6);