美文网首页
Cupy与Numpy的数据类型互转

Cupy与Numpy的数据类型互转

作者: 胜负55开 | 来源:发表于2019-10-24 18:41 被阅读0次

用cupy和numpy直接创建的数组/矩阵,类型是不一样的!无法相互之间进行运算。

用type查看二者的数据类型:

  • cupy的数据类型:cupy.core.core.ndarray
  • numpy的数据类型:numpy.ndarray

其实,如果只需要进行大量数组/矩阵间计算,然后将结果存储到某个"容器"之中,以cupy目前实现的函数功能来看,只用cupy来完成肯定是够的,完全可以一点没有numpy的影子。

但是,很多的"第三方包是不认识cupy的数据类型"的!本人就发现"matplotlib"就不识别cupy的数据类型!因此,如果除"纯计算"外还有别的需求,需要将cupy数据先转为numpy数据(numpy是转不了cupy的!)

cupy转numpy的函数:cupy.asnumpy()

import cupy as cp
import numpy as np

x = cp.ones( (1024,512,4,4) )*1024.
x1 = cp.asnumpy(x)  # 直接cupy数据类型转numpy

print( type(x) )
print( type(x1) )

结果:

<class 'cupy.core.core.ndarray'>
<class 'numpy.ndarray'>

有了这么简单的数据互转函数,只要计算量大,可以用cupy完全替换numpy!
直到需要使用其他第三方包的时候,再把cupy数据类型转回到numpy即可。

相关文章

  • Cupy与Numpy的数据类型互转

    用cupy和numpy直接创建的数组/矩阵,类型是不一样的!无法相互之间进行运算。 用type查看二者的数据类型:...

  • cupy数组赋值

    给数组进行初始化时,numpy确实要比cupy好用!因为numpy完全认识python自带的各种数据类型,如列表、...

  • CuPy内存管理

    CuPy是CUDA加速的NumPy兼容矩阵库,可以直接把Numpy的数组转为CuPy,提高运算速度。 内存管理 默...

  • [Python]高阶-玩转数据类型转换

    一、Python、NumPy、Pandas数据类型互转 行为输入列为输出————>strlisttuplesetd...

  • iOS-类型转换

    数据类型与char*类型的相互转换(NSData <-> char *) 字符串类型与数据类型的相互转换(NSSt...

  • Cupy的用处概述

    前提:传统的数组和矩阵都是通过numpy来设定,然后numpy来调用cpu计算!cupy的作用:数组和矩阵都是通过...

  • cupy或numpy中"数组"与"矩阵"的区别

    前言: cupy和numpy在矩阵和数组的使用上没有区别,在认识/创建层面有一些区别。 numpy中:任何创建函数...

  • 【Python图像处理】RGB颜色转HSV颜色的快速实现

    传送门 rgb转hsv的简单实现 rgb与hsv互转的数学公式 思路   使用NumPy。NumPy对数组和矩阵的...

  • Numpy与图像互转

    图像转numpy 这相对简单,示例如下: numpy转图像 现假设变量img为np.array数据,则将其转为图像...

  • NumPy 超详细教程(2):数据类型

    NumPy 数据类型 1、NumPy 中的数据类型 NumPy 支持比 Python 更多种类的数值类型,下表所列...

网友评论

      本文标题:Cupy与Numpy的数据类型互转

      本文链接:https://www.haomeiwen.com/subject/cwvdvctx.html