list转torch tensor

最新推荐文章于 2024-08-29 09:53:07 发布

AI算法网奇

最新推荐文章于 2024-08-29 09:53:07 发布

阅读量7.4w

点赞数 5

分类专栏： pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/82828221

版权

pytorch知识宝典专栏收录该内容

500 篇文章 227 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了如何将list转换为torch Tensor，指出直接转换可能较慢，推荐使用numpy转换以提高效率。当数据维度不匹配时，会抛出错误：ValueError: expected sequence of length 3 at dim 1 (got 1)。解决方法是确保数据对齐，例如3*3*3,5*3*2,7*8等格式。" 48323861,4997259,Java编程：多线程与面向对象设计模式解析,"['Java学习', '多线程', '面向对象设计思想', '设计模式']

摘要由CSDN通过智能技术生成

list直接转tensor，这种的可能会慢：

import torch
pick=[1,2,3,4]
bbb=torch.tensor(pick, dtype=torch.long)
print(bbb)

numpy转化的比较快

import cv2
import torch
import time

img = cv2.imread('d:/nv.jpg')
start=time.time()
# tensor = torch.Tensor(img)
tensor= torch.from_numpy(img)

print("cost time",time.time()-start,tensor.shape)

如果数据维度没有对齐，会报错：

import cv2
import torch
import time

start=time.time()

data_a=[[1,2,3],[2,3,4],[1]]

data_b=torch.Tensor(data_a)


print("cost time",time.time()-start,data_b.shape)

异常：

Traceback (mo