Python同模块数据类型转换

gy笨瓜

于 2020-10-10 22:54:12 发布

阅读量303

点赞数

分类专栏：数据类型转换文章标签：数据类型转换 PyTorch numpy

本文链接：https://blog.csdn.net/u012633319/article/details/109005722

版权

数据类型转换专栏收录该内容

2 篇文章 0 订阅

订阅专栏

主要涉及模块：
PyTorch，numpy

import torch
import numpy as np

# img_np,img_torch_tensor分别表示numpy.ndarray,torch.Tensor对象

# 1.numpy.ndarray转换元素类型为np.float32（可替换为numpy允许的数据类型）
img_np = np.float32(img_np)

# 2.torch.Tensor转换元素类型为torch.float32（可替换为torch允许的数据类型）
img_torch_tensor = img_torch_tensor.type(torch.float32)
img_torch_tensor = img_torch_tensor.to(torch.float32)