数据类型包括Python数据类型、Numpy数据类型和Pytorch中的tensor,Pytorch中的tensor又包括CPU上的数据类型和
GPU上的数据类型。
一、Python数据类型
Python3 中有六个标准的数据类型:
* Number(数字)
* String(字符串)
* List(列表)
* Tuple(元组)
* Set(集合)
* Dictionary(字典)
在三种数据类型转换中,我们最常用到的就是Number中的int类型,和List类型。
二、Numpy数据类型
numpy 支持的数据类型比 Python 内置的类型要多很多,基本上可以和 C 语言的数据类型对应上,其中部分类型对应为 Python
内置的类型。下表列举了常用 NumPy 基本类型。
三、Tensor数据类型
在torch中CPU和GPU张量分别有8种数据类型, GPU tensor比CPU tensor多了.cuda
四、 不同数据类型之间的转换
python基本数据类型和Numpy数据类型之间的转换:
Numpy--->python 使用data.item(),data为numpy变量
python--->Numpy 使用np.array(data),data为python变量
Tensor与numpy之间的转换:
Tensor---->Numpy 使用 data.numpy(),data为Tensor变量
Numpy ----> Tensor 使用torch.from_numpy(data),data为numpy变量
GPU tensor与CPUtensor之间的转换:
CPU张量 ----> GPU张量, 使用data.cuda()
GPU张量 ----> CPU张量 使用data.cpu()