add unchecked getters

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "micro_ndarray"
-version = "0.2.3"
+version = "0.2.4"
 edition = "2021"
 repository = "https://github.com/tudbut/micro_ndarray"
 license = "MIT"

src/array.rs

@@ -1,4 +1,7 @@
-use std::{ops::{Index, IndexMut}, slice};
+use std::{
+    ops::{Index, IndexMut},
+    slice,
+};
 
 use crate::iterator::Iter;
 
@@ -105,6 +108,22 @@ impl<'a, T, const D: usize> Array<T, D> {
         }
     }
 
+    pub unsafe fn get_unchecked(&'a self, loc: [usize; D]) -> &'a T {
+        let mut real_loc = 0;
+        for (i, dim) in loc.iter().enumerate() {
+            let mut dim = *dim;
+            if i == 0 {
+                real_loc += dim;
+                continue;
+            }
+            for s in &self.size[0..i] {
+                dim *= s;
+            }
+            real_loc += dim;
+        }
+        self.data.get_unchecked(real_loc)
+    }
+
     pub fn get_mut(&'a mut self, loc: [usize; D]) -> Option<&'a mut T> {
         self.internal_get_mut(loc, false)
     }
@@ -141,6 +160,22 @@ impl<'a, T, const D: usize> Array<T, D> {
         }
     }
 
+    pub unsafe fn get_unchecked_mut(&'a mut self, loc: [usize; D]) -> &'a mut T {
+        let mut real_loc = 0;
+        for (i, dim) in loc.iter().enumerate() {
+            let mut dim = *dim;
+            if i == 0 {
+                real_loc += dim;
+                continue;
+            }
+            for s in &self.size[0..i] {
+                dim *= s;
+            }
+            real_loc += dim;
+        }
+        self.data.get_unchecked_mut(real_loc)
+    }
+
     pub fn iter(&mut self) -> Iter<slice::Iter<T>, D> {
         Iter::new(self)
     }