Add transpose, horzcat, vertcat and Fix repmat threadPerBlock problem.

2023-11-30 10:22:06 +08:00
parent 87dd34095c
commit a8c65f21b1
3 changed files with 291 additions and 18 deletions
--- a/src/Function1D.cuh
+++ b/src/Function1D.cuh
@@ -63,6 +63,12 @@ namespace Aurora

    float norm(const CudaMatrix& aMatrix, NormMethod aNormMethod);

+    CudaMatrix transpose(const CudaMatrix& aMatrix);
+
+    CudaMatrix horzcat(const CudaMatrix& aMatrix1, const CudaMatrix& aMatrix2);
+
+    CudaMatrix vertcat(const CudaMatrix& aMatrix1, const CudaMatrix& aMatrix2);
+
    // ------compareSet----------------------------------------------------