利用CUDA計算向量與矩陣每一行的歐式距離

2021-10-11 11:31:00 字數 999 閱讀 8848

本文作為筆者的學習筆記,**僅供參考。

**實現:計算向量a(n維)到矩陣b(n*n)每一行的歐式距離,並將結果輸出到向量c(n維)。

其中,向量a與矩陣b中的元素均規定為整數,輸出的向量c中資料型別為浮點數。

具體如下:

#include

#include

const

int block_size =5;

const

int grid_size =5;

#define n 10

__global__ void

dist

(int

*a,int

*b,float

*c) c[blockidx.x]

=sqrt

(t);

}int

main

(void)}

cudamemcpy

(d_a,a,size,cudamemcpyhosttodevice)

;cudamemcpy

(d_b,b,size*n,cudamemcpyhosttodevice);

dim3 threadblock

(block_size,block_size)

; dim3 grid

(grid_size,grid_size)

; dist<<

>

>

(d_a,d_b,d_c)

;cudamemcpy

(c,d_c,n*

sizeof

(float

),cudamemcpydevicetohost)

;for

(i=0

;i)cudafree

(d_a)

;cudafree

(d_b)

;cudafree

(d_c)

;return0;

}

**尚待完善,目前發現的不足如下:

3 3numpy 向量與矩陣的計算,矩陣的逆

import numpy numpy.array中的運算 給定乙個向量,讓向量中每乙個數乘以2 a 0,1,2 a 2 0,2,4 n 10 l i for i in range n a for e in l 方法2 利用生成表示式的方式處理,效率會變高 a 2 e for e in l l num...

類與物件(一) 6 矩陣計算

問題描述 矩陣是線性代數中的重要概念,應用領域非常廣泛,在c c 中,通常將矩陣定義為乙個二維陣列。本問題中,將輸入兩個矩陣 a 和 b,實現對矩陣的數乘 矩陣加法 矩陣乘法以及行列式的計算。如果對矩陣的演算法不了解,請查閱相關資料。輸入形式 輸入的第一行為兩個正整數 m 和 n,分別表示矩陣 a ...

numpy 基礎 陣列與向量計算(一)

在本次的部落格中,我將展示一些 jupyter陣列與向量計算的常用語法。import numpy as np my arr np.arange 10000 生成範圍為 0,9999 這10000個數字 my list list range 10000 time for i in range 10 m...