Numpy和Pytorch中数据类型的转换

最新推荐文章于 2023-01-19 17:30:21 发布

Huang_Fj

最新推荐文章于 2023-01-19 17:30:21 发布

阅读量1.6k

点赞数

分类专栏： pytorch numpy基础文章标签： pytorch 深度学习神经网络

本文链接：https://blog.csdn.net/Huang_Fj/article/details/120798943

版权

数据类型的转换

numpy：numpy中的数据类型详见：https://numpy.org/doc/stable/reference/arrays.dtypes.html
可以直接用astype()对其进行转换
torch：torch中的数据类型详见：
https://pytorch.org/docs/stable/tensors.html
对于Tensor类型的数据，可以直接在其后加上.long()，.float()，.double()，.int()等。也可以利用.to()，里面的参数可以是dtype包含的数据类型，也可以是cpu或者cuda。

numpy中默认的数据类型是float64，而torch中默认的数据类型是float32。
举例1：

import numpy as np
arr = np.random.randn(2,2)
arr_float32 = arr.astype(np.float32)
print(arr.dtype, arr_float32.dtype)
#>>> dtype('float64') dtype('float32')

import torch
tensor = torch.randn(2,2)
tensor_long = tensor.long()
tensor_double = tensor.to