Execute tensor product functions with RowFn

vortex-tensor/src/scalar_fns/arithmetic.rs

@@ -4,7 +4,8 @@
 //! Row-level arithmetic shared by tensor scalar functions.
 //!
 //! These kernels preserve the scalar functions' left-to-right floating-point arithmetic. Keeping
-//! the arithmetic here ensures that fused and single-result operations use the same result contract.
+//! the arithmetic here ensures that fused and single-result operations use the same result
+//! contract.
 
 use num_traits::Float;
 use vortex_array::dtype::NativePType;
@@ -21,3 +22,14 @@ pub(crate) fn l2_norm_row<T: Float + NativePType>(row: &[T]) -> T {
 
     sum_squared.sqrt()
 }
+
+/// Computes `sum(lhs_i * rhs_i)` for one pair of rows.
+///
+/// Both rows must have the same length. Otherwise, trailing values are ignored and the result is
+/// incorrect. Callers preserve the left-to-right multiplication and accumulation order.
+pub(crate) fn inner_product_row<T: Float + NativePType>(lhs: &[T], rhs: &[T]) -> T {
+    lhs.iter()
+        .zip(rhs)
+        .map(|(&lhs_value, &rhs_value)| lhs_value * rhs_value)
+        .fold(T::zero(), |sum, product| sum + product)
+}