Pytorch中数据的类型及转换

LYS_1129

于 2020-04-24 17:05:11 发布

阅读量710

点赞数

CC 4.0 BY-SA版权

分类专栏：深度学习 Python 文章标签：深度学习 python

本文链接：https://blog.youkuaiyun.com/weixin_43915511/article/details/105734689

本文探讨了Pytorch框架中涉及的数据类型，包括Numpy、Tensor、CUDA和CPU类型数据，并通过波士顿房价预测数据集展示了这些类型间的转换实践。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在用Pytorch框架设计神经网络训练时，会涉及到数据的类型问题。按训练方式不同，数据可以分为基于CPU的和基于GPU的。调用不同库初始化时，又可大致分为numpy类型和tensor类型。以下结合项目实际，通过代码来对比这些数据类型，并实现它们之间的相互转换。
我们以波士顿房价预测数据集来讨论说明。

from sklearn.datasets import load_boston
dataset=load_boston()
x_data=dataset.data
y_data=dataset.target
n_data=dataset.feature_names

1.Numpy类型数据

print

最低0.47元/天解锁文章