SimpleJCublas.java

    public static INDArray gemv(INDArray A, INDArray B, INDArray C, float alpha, float beta) {


        JCublasNDArray cA = (JCublasNDArray) A;
        JCublasNDArray cB = (JCublasNDArray) B;
        JCublasNDArray cC = (JCublasNDArray) C;

       char trans = 'n';
        if (A.rows() == B.columns()) {
            trans = 'T';
        }
        JCublas.cublasDgemv(
                'n', //trans
                A.rows(),  // m
                A.columns(), // n
                alpha, //alpha
               cA.pointer(), // A
                A.rows(),  // lda
                cB.pointer(), // x
                B.rows(), // incx
                beta,  // beta
                cC.pointer(), // y
                1); // incy

       cC.getData();



        return C;
    }