Py学习  »  Python

将python列表转换为pytorch张量

Andrea Eunbee Jang • 5 年前 • 1247 次点击  

将python列表转换为PyTorch张量的传统方法是什么?

a = [0,0]
b = [1,1]
c = [2]
c = [a, b, c]

我想把c转换成一个扁平的火炬张量,如下所示:

torch([ 0, 0, 1, 1, 2])
Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/53439
 
1247 次点击  
文章 [ 1 ]  |  最新文章 5 年前