rvcmmul
实数向量与复数矩阵的行或列乘法运算。
函数接口声明如下:
float型函数操作:
void (vsip_rvcmmul_f)(const vsip_vview_f *a, const vsip_cmview_f *B, vsip_major major, const vsip_cmview_f *R);
参数
参数名 |
描述 |
取值范围 |
输入/输出 |
---|---|---|---|
a |
指向源实数向量的指针。 |
非空 |
输入 |
B |
指向源复数矩阵的指针。 |
非空 |
输入 |
major |
按行或列向量运算。 |
枚举vsip_major,选项: VSIP_ROW:行运算 VSIP_COL:列运算 |
输入 |
R |
指向目标实数矩阵的指针。 |
非空 |
输出 |
异常输入
空指针输入时,函数直接返回。
示例
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 | #include <stdio.h> #include <stdlib.h> #include <stdint.h> #include "kvsip.h" #include "vsip.h" #include "vsip_type.h" void RvcmmulExample() { int32_t row = 4; int32_t col = 3; int64_t rowStride = 1; int64_t colStride = row * rowStride; uint64_t offset = 0; vsip_major major = VSIP_ROW; int32_t vLen = row * rowStride + offset; int32_t cmLen = 2 * (col * colStride + offset); float *src1 = (float *)malloc(vLen * sizeof(float)); float *src2 = (float *)malloc(cmLen * sizeof(float)); float *dst = (float *)malloc(cmLen * sizeof(float)); for (int32_t i = 0; i < vLen; ++i) { src1[i] = 1 + i / 100.0f; } for (int32_t i = 0; i < cmLen; ++i) { src2[i] = 2 + i / 100.0f; } vsip_block_f *block_a = vsip_blockbind_f(src1, vLen, VSIP_MEM_NONE); vsip_cblock_f *block_b = vsip_cblockbind_f(src2, NULL, cmLen, VSIP_MEM_NONE); vsip_cblock_f *block_r = vsip_cblockbind_f(dst, NULL, cmLen, VSIP_MEM_NONE); vsip_vview_f *a = vsip_vbind_f(block_a, offset, rowStride, vLen); vsip_blockadmit_f(block_a, VSIP_TRUE); vsip_cmview_f *B = vsip_cmbind_f(block_b, offset, colStride, col, rowStride, row); vsip_cblockadmit_f(block_b, VSIP_TRUE); vsip_cmview_f *R = vsip_cmbind_f(block_r, offset, colStride, col, rowStride, row); vsip_cblockadmit_f(block_r, VSIP_TRUE); vsip_rvcmmul_f(a, B, major, R); vsip_valldestroy_f(a); vsip_cmalldestroy_f(B); vsip_cmalldestroy_f(R); printf("src1:\n"); for (int32_t i = 0; i < row; ++i) { printf("%.5f ", src1[i]); } printf("\n\nsrc2:\n"); for (int32_t i = 0; i < col; ++i) { for (int32_t j = 0; j < row; ++j) { printf("%.5f ", src2[i * row * 2 + j * 2]); printf("%.5f ", src2[i * row * 2 + j * 2 + 1]); } printf("\n"); } printf("\ndst:\n"); for (int32_t i = 0; i < col; ++i) { for (int32_t j = 0; j < row; ++j) { printf("%.5f ", dst[i * row * 2 + j * 2]); printf("%.5f ", dst[i * row * 2 + j * 2 + 1]); } printf("\n"); } free(src1); free(src2); free(dst); } int main(void) { RvcmmulExample(); return 0; } |
运行结果:
1 2 3 4 5 6 7 8 9 10 11 12 | src1: 1.00000 1.01000 1.02000 1.03000 src2: 2.00000 2.01000 2.02000 2.03000 2.04000 2.05000 2.06000 2.07000 2.08000 2.09000 2.10000 2.11000 2.12000 2.13000 2.14000 2.15000 2.16000 2.17000 2.18000 2.19000 2.20000 2.21000 2.22000 2.23000 dst: 2.00000 2.01000 2.04020 2.05030 2.08080 2.09100 2.12180 2.13210 2.08000 2.09000 2.12100 2.13110 2.16240 2.17260 2.20420 2.21450 2.16000 2.17000 2.20180 2.21190 2.24400 2.25420 2.28660 2.29690 |
父主题: 矩阵运算函数