ndarray 与 pytorch tensor , variable 相互转化

最新推荐文章于 2022-03-01 15:54:20 发布

JChowCUG

最新推荐文章于 2022-03-01 15:54:20 发布

阅读量1.3k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： pytorch 习惯养成 Deep Learning

本文链接：https://blog.youkuaiyun.com/JJChowLoT/article/details/96317023

习惯养成同时被 3 个专栏收录

24 篇文章

订阅专栏

Deep Learning

15 篇文章

订阅专栏

pytorch

14 篇文章

订阅专栏

本文介绍了如何在Python中进行不同数据类型之间的转换，包括从list到ndarray、从ndarray到tensor、从tensor到Variable等常见操作，并展示了如何访问和修改Variable的数据。

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

import torch
import numpy as np
from torch.autograd import Variable

list_1 = [1, 2, 3, 4, 5]
# list to ndarray
ndarray_1 = np.array(list_1)
# ndarray to list
list_2 = ndarray_1.tolist()


ndarray_2 = np.array([1, 2, 3, 4, 5])
# ndarray to tensor
tensor_1 = torch.from_numpy(ndarray_2)
# tensor to ndarray
ndarray_3 = tensor_1.numpy()


# numpy to Variable
variable_1 = Variable(torch.from_numpy(ndarray_3))
# variable to numpy
ndarray_4 = variable_1.numpy()

# 访问修改variable.data
variable_1.data = torch.from_numpy(np.random.uniform(2, 3, size=(5, 1)))