Numpy 是强大的数据操作工具,但不能在GPU上运行,需将ndarray转换成tensor才行。开始前先搞清楚需要哪种数据类型,再进行转换。本文简述list、ndarray和tensor的常用操作。
1.基本转换图
2. 数据类型
ndarray和tensor:查看dtype参数即可。具体查看pytorch官网及菜鸟教程介绍部分。
list需要特别留意每个item类型,将数据转换成可以直接使用的类型,并进行存储。
(1)tensor数据类型
(2)ndarray数据类型
Numpy 是强大的数据操作工具,但不能在GPU上运行,需将ndarray转换成tensor才行。开始前先搞清楚需要哪种数据类型,再进行转换。本文简述list、ndarray和tensor的常用操作。
ndarray和tensor:查看dtype参数即可。具体查看pytorch官网及菜鸟教程介绍部分。
list需要特别留意每个item类型,将数据转换成可以直接使用的类型,并进行存储。
(1)tensor数据类型
(2)ndarray数据类型