Add repmat and repmat3d, Fix sqrt with complex.

2023-11-27 09:47:59 +08:00
parent 3d68171394
commit fc9b6be9e8
4 changed files with 161 additions and 2 deletions
--- a/src/Function1D.cuh
+++ b/src/Function1D.cuh
@@ -39,6 +39,12 @@ namespace Aurora
    CudaMatrix sign(const CudaMatrix& aMatrix);

    CudaMatrix sign(const CudaMatrix&& aMatrix);
+
+    CudaMatrix repmat(const CudaMatrix& aMatrix,int aRowTimes, int aColumnTimes);
+
+    CudaMatrix repmat(const CudaMatrix& aMatrix,int aRowTimes, int aColumnTimes, int aSliceTimes);
+
+    CudaMatrix repmat3d(const CudaMatrix& aMatrix,int aRowTimes, int aColumnTimes, int aSliceTimes);
 }

 #endif //AURORA_CUDA_FUNCTION1D_H