feat: add pow method

Author: siliconlad

src/matrix.rs

@@ -1,6 +1,6 @@
 use std::ops::{Add, Sub, Mul, Div, Deref, Index, IndexMut, DerefMut};
 
-use num::Num;
+use num::{Num, Float};
 use crate::error::ShapeError;
 use crate::shape;
 use crate::coord;
@@ -90,6 +90,12 @@ impl<T: Num + PartialOrd + Copy> DynamicMatrix<T> {
     }
 }
 
+impl<T: Float + PartialOrd + Copy> DynamicMatrix<T> {
+    pub fn pow(&self, power: T) -> DynamicMatrix<T> {
+        DynamicMatrix::from_tensor(self.tensor.pow(power)).unwrap()
+    }
+}
+
 // Scalar Addition
 impl<T: Num + PartialOrd + Copy> Add<T> for DynamicMatrix<T> {
     type Output = DynamicMatrix<T>;
@@ -606,4 +612,17 @@ mod tests {
         assert_eq!(matrix[coord![1, 0]], 3.0);
         assert_eq!(matrix[coord![1, 1]], 4.0);
     }
+
+    #[test]
+    fn test_pow_matrix() {
+        let shape = shape![2, 2].unwrap();
+        let data = vec![2.0, 3.0, 4.0, 5.0];
+        let matrix = DynamicMatrix::new(&shape, &data).unwrap();
+        let result = matrix.pow(2.0);
+        assert_eq!(result[coord![0, 0]], 4.0);
+        assert_eq!(result[coord![0, 1]], 9.0);
+        assert_eq!(result[coord![1, 0]], 16.0);
+        assert_eq!(result[coord![1, 1]], 25.0);
+        assert_eq!(result.shape(), &shape);
+    }
 }

src/tensor.rs

@@ -1,4 +1,4 @@
-use num::Num;
+use num::{Num, Float};
 use std::ops::{Add, Sub, Mul, Div};
 
 use crate::shape;
@@ -253,6 +253,16 @@ impl<T: Num + PartialOrd + Copy> Tensor<T> {
     }
 }
 
+impl<T: Float + PartialOrd + Copy> Tensor<T> {
+    pub fn pow(&self, power: T) -> Tensor<T> {
+        let mut result = Tensor::zeros(&self.shape);
+        for i in 0..self.size() {
+            result.data[i] = self.data[i].clone().powf(power);
+        }
+        result
+    }
+}
+
 // Element-wise Multiplication
 impl<T: Num + PartialOrd + Copy> Mul<T> for Tensor<T> {
     type Output = Tensor<T>;
@@ -1205,5 +1215,35 @@ mod tests {
         let display = tensor.display();
         assert_eq!(display, "[[[[1, 2],\n   [3, 4]],\n\n  [[5, 6],\n   [7, 8]]],\n\n\n [[[9, 10],\n   [11, 12]],\n\n  [[13, 14],\n   [15, 16]]]]");
     }
+
+    #[test]
+    fn test_pow_tensor_square() {
+        let shape = shape![2, 2].unwrap();
+        let data = vec![1.0, 2.0, 3.0, 4.0];
+        let tensor = Tensor::new(&shape, &data).unwrap();
+        let result = tensor.pow(2.0);
+        assert_eq!(result.shape(), &shape);
+        assert_eq!(result.data, DynamicStorage::new(vec![1.0, 4.0, 9.0, 16.0]));
+    }
+
+    #[test]
+    fn test_pow_tensor_sqrt() {
+        let shape = shape![2, 2].unwrap();
+        let data = vec![1.0, 4.0, 9.0, 16.0];
+        let tensor = Tensor::new(&shape, &data).unwrap();
+        let result = tensor.pow(0.5);
+        assert_eq!(result.shape(), &shape);
+        assert_eq!(result.data, DynamicStorage::new(vec![1.0, 2.0, 3.0, 4.0]));
+    }
+
+    #[test]
+    fn test_pow_tensor_negative_exponent() {
+        let shape = shape![2, 2].unwrap();
+        let data = vec![1.0, 2.0, 4.0, 8.0];
+        let tensor = Tensor::new(&shape, &data).unwrap();
+        let result = tensor.pow(-1.0);
+        assert_eq!(result.shape(), &shape);
+        assert_eq!(result.data, DynamicStorage::new(vec![1.0, 0.5, 0.25, 0.125]));
+    }
 }
 

src/vector.rs

@@ -1,6 +1,7 @@
 use std::ops::{Add, Sub, Mul, Div, Deref, Index, IndexMut, DerefMut};
 
 use num::Num;
+use num::Float;
 use crate::error::ShapeError;
 use crate::shape;
 use crate::coord;
@@ -84,6 +85,12 @@ impl<T: Num + PartialOrd + Copy> DynamicVector<T> {
     }
 }
 
+impl<T: Float + PartialOrd + Copy> DynamicVector<T> {
+    pub fn pow(&self, power: T) -> DynamicVector<T> {
+        DynamicVector::from_tensor(self.tensor.pow(power)).unwrap()
+    }
+}
+
 // Scalar Addition
 impl<T: Num + PartialOrd + Copy> Add<T> for DynamicVector<T> {
     type Output = DynamicVector<T>;
@@ -585,4 +592,17 @@ mod tests {
         assert_eq!(result[3], 2.0);
         assert_eq!(result.shape(), &shape);
     }
+
+    #[test]
+    fn test_pow_vector() {
+        let shape = shape![4].unwrap();
+        let data = vec![2.0, 3.0, 4.0, 5.0];
+        let vector = DynamicVector::new(&data).unwrap();
+        let result = vector.pow(2.0);
+        assert_eq!(result[0], 4.0);
+        assert_eq!(result[1], 9.0);
+        assert_eq!(result[2], 16.0);
+        assert_eq!(result[3], 25.0);
+        assert_eq!(result.shape(), &shape);
+    }
 }