diff --git a/src/Function3D.cpp b/src/Function3D.cpp
index cfb24e4..ddb5fc8 100644
--- a/src/Function3D.cpp
+++ b/src/Function3D.cpp
@@ -117,3 +117,8 @@ Matrix Aurora::size(const Matrix &aMatrix)
         return Matrix::New(output,2,1,1);
     }
 }
+
+int Aurora::size(const Matrix &aMatrix,int dims)
+{
+    return aMatrix.getDimSize(dims-1);
+}
diff --git a/src/Function3D.h b/src/Function3D.h
index bf63753..17bc282 100644
--- a/src/Function3D.h
+++ b/src/Function3D.h
@@ -42,6 +42,7 @@ namespace Aurora {
     Matrix interpn(const Matrix& aX, const Matrix& aY, const Matrix& aZ, const Matrix& aV, const Matrix& aX1, const Matrix& aY1, const Matrix& aZ1,InterpnMethod aMethod);
 
     Matrix size(const Matrix &aMatrix);
+    int size(const Matrix &aMatrix,int dims);
 };