EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助

?omatmul

矩阵缩放后out-place乘计算。

op可取值:

接口定义

C interface:

void cblas_somatmul(const enum CBLAS_ORDER ordering, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB, const BLASINT m, const BLASINT n, const float alpha, float *A, const BLASINT lda, float *B, const BLASINT ldb, float *C, const BLASINT ldc);

void cblas_domatmul(const enum CBLAS_ORDER ordering, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB, const BLASINT m, const BLASINT n, const double alpha, double *A, const BLASINT lda, double *B, const BLASINT ldb, double *C, const BLASINT ldc);

void cblas_comatmul(const enum CBLAS_ORDER ordering, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB, const BLASINT m, const BLASINT n, const float *alpha, float *A, const BLASINT lda, float *B, const BLASINT ldb, float *C, const BLASINT ldc);

void cblas_zomatmul(const enum CBLAS_ORDER ordering, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB, const BLASINT m, const BLASINT n, const double *alpha, double *A, const BLASINT lda, double *B, const BLASINT ldb, double *C, const BLASINT ldc);

Fortran interface:

CALL SOMATMUL(ORDER, TRANSA, TRANSB, M, N, ALPHA, A, LDA, B, LDB, C, LDC)

CALL DOMATMUL(ORDER, TRANSA, TRANSB, M, N, ALPHA, A, LDA, B, LDB, C, LDC)

CALL COMATMUL(ORDER, TRANSA, TRANSB, M, N, ALPHA, A, LDA, B, LDB, C, LDC)

CALL ZOMATMUL(ORDER, TRANSA, TRANSB, M, N, ALPHA, A, LDA, B, LDB, C, LDC)

参数

参数名

类型

描述

输入/输出

order

枚举类型CBLAS_ORDER

表示矩阵是行主序或列主序。

输入

TransA

枚举类型CBLAS_TRANSPOSE

矩阵A为常规矩阵,转置矩阵或共轭矩阵。

  • 如果TransA = CblasNoTrans,op(A) = A。
  • 如果TransA = CblasTrans,op(A) = A'。
  • 如果TransA = CblasConjNoTrans,op(A) = conj(A)。
  • 如果TransA = CblasConjTrans,op(A) = conj(A')。

输入

TransB

枚举类型CBLAS_TRANSPOSE

矩阵B为常规矩阵,转置矩阵或共轭矩阵。

  • 如果TransB = CblasNoTrans,op(B) = B。
  • 如果TransB = CblasTrans,op(B) = B'。
  • 如果TransB = CblasConjNoTrans,op(B) = conj(B)。
  • 如果TransB = CblasConjTrans,op(B) = conj(B')。

输入

M

整型数

矩阵op(A)、op(B)、C的行。

输入

N

整型数

矩阵op(A)、op(B)、C的列。

输入

alpha

  • 在somatmul中是单精度浮点类型。
  • 在domatmul中是双精度浮点类型。
  • 在comatmul中是单精度复数类型。
  • 在zomatmul中是双精度复数类型。

表示乘法系数。

输入

A

  • 在somatmul中是单精度浮点类型。
  • 在domatmul中是双精度浮点类型。
  • 在comatmul中是单精度复数类型。
  • 在zomatmul中是双精度复数类型。

矩阵A。

输入

lda

整型数

  • 矩阵A为列存,TransA = CblasNoTrans,lda至少max(1, m),否则max(1, n)。
  • 矩阵A为行存,TransA = CblasNoTrans,lda至少max(1, n),否则max(1, m)。

输入

B

  • 在somatmul中是单精度浮点类型。
  • 在domatmul中是双精度浮点类型。
  • 在comatmul中是单精度复数类型。
  • 在zomatmul中是双精度复数类型。

矩阵B。

输入

ldb

整型数

  • 矩阵为列存,TransB = CblasNoTrans,ldb至少max(1, m),否则max(1, n)。
  • 矩阵为行存,TransB = CblasNoTrans,ldb至少max(1, n),否则max(1, m)。

输入

C

  • 在somatmul中是单精度浮点类型。
  • 在domatmul中是双精度浮点类型。
  • 在comatmul中是单精度复数类型。
  • 在zomatmul中是双精度复数类型。

矩阵C。

输出

ldc

整型数

矩阵为列存,ldc至少max(1, m),否则max(1, n)。

输入

依赖

#include "kblas.h"

示例

C interface:

    int m = 4;
    int n = 3;
    int lda = 4;
    int ldb = 4;
    int ldc = 4;
    float alpha = 2.0;

    /** 
     *    A: 
     *         -0.399988, -0.211286, 0.226351,           
     *         0.057980, 0.185737, 0.015954,     
     *         -0.181192, 0.092167, -0.125647,     
     *         0.206140, -0.441963, -0.307887       
     *    B: 
     *         0.274879, -0.184964, -0.331672,              
     *         -0.476422, 0.318246, 0.295214,     
     *         -0.067229, 0.314890, -0.072014,     
     *         -0.088924, -0.222191, -0.121713,     
     *    C: 
     *         0, 0, 0,
     *         0, 0, 0,
     *         0, 0, 0,
     *         0, 0, 0
     */

    float a[12] = {-0.399988, -0.211286, 0.226351,
                   0.057980, 0.185737, 0.015954,
                   -0.181192, 0.092167, -0.125647,
                   0.206140, -0.441963, -0.307887
    };

    float b[12] = {0.274879, -0.184964, -0.331672,
                   -0.476422, 0.318246, 0.295214,
                   -0.067229, 0.314890, -0.072014,
                   -0.088924, -0.222191, -0.121713,
    };

    float c[12] = {0};

    cblas_somatmul(CblasColMajor, CblasNoTrans, CblasNoTrans, m, n, alpha, a, lda, b, ldb, c, ldc);

    /** 
     * Output C: 
     *        -0.219897, 0.078161, -0.150148,      
     *        -0.055246, 0.118220, 0.009419,
     *        0.024363, 0.058045, 0.018097, 
     *        -0.036661, 0.196401, 0.074947
     */

Fortran interface:

      PROGRAM OMATMUL
      INTEGER :: M=4, N=3
      INTEGER :: LDA=4, LDB=4, LDC=4
      REAL(4) :: ALPHA=2.0
      REAL(4) :: A(4, 3), B(4, 3), C(4, 3)
      DATA A/-0.399988, -0.211286, 0.226351,
             0.057980, 0.185737, 0.015954,
             -0.181192, 0.092167, -0.125647,
             0.206140, -0.441963, -0.307887/
      DATA B/0.274879, -0.184964, -0.331672,
             -0.476422, 0.318246, 0.295214,
             -0.067229, 0.314890, -0.072014,
             -0.088924, -0.222191, -0.121713/
      DATA C/0.0, 0.0, 0.0,
             0.0, 0.0, 0.0,
             0.0, 0.0, 0.0/
      EXTERNAL SOMATMUL
      CALL SOMATMUL('C', 'N', 'N', M, N, ALPHA, A, LDA, B, LDB, C, LDC)
      END

     * Output C: 
     *        -0.219897, 0.078161, -0.150148,      
     *        -0.055246, 0.118220, 0.009419,
     *        0.024363, 0.058045, 0.018097,
     *        -0.036661, 0.196401, 0.074947