當前位置:成語大全網 - 書法字典 - Cupy和Numpy之間的數據類型轉換

Cupy和Numpy之間的數據類型轉換

cupy和numpy直接創建的數組/矩陣的類型是不同的!不能互相操作。

使用type查看其數據類型:

事實上,如果只需要數組/矩陣之間的大量計算,然後將結果存儲在壹個“容器”中,從cupy目前實現的功能來看,只使用cupy肯定是足夠的,完全沒有numpy的影子。

但是,很多“第三方包不知道cupy的數據類型”!我發現“matplotlib”不能識別cupy的數據類型!因此,如果除了“純計算”之外還有其他要求,則需要先將cupy數據轉換為NUPY數據(NUPY不能轉換為cupy!)

從cupy到numpy的函數:cupy.asnumpy()

結果:

這麽簡單的數據轉換功能,只要計算量大,完全可以用cupy代替numpy!

在需要使用其他第三方包之前,可以將cupy數據類型改回numpy。