list转torch tensor

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

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

numpy转化的比较快

import cv2
import torch
import timeimg = 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 timestart=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 (most recent call last):

File "E:/github/YOLOv3_PyTorch/common/__init__.py", line 4, in b=torch.Tensor(a) ValueError: expected sequence of length 3 at dim 1 (got 1)

解决方&

list转torch tensor相关推荐

PyTorch 笔记（02）— 常用创建 Tensor 方法（torch.Tensor、ones、zeros、eye、arange、linspace、rand、randn、new）
1. Tensor 概念分类 PyTorch 中的张量(Tensor)类似 NumPy 中的 ndarrays,之所以称之为 Tensor 的另一个原因是它可以运行在 GPU 中,以加速运算. 1.1 ...
pytorch中的torch.tensor.repeat以及torch.tensor.expand用法
文章目录 torch.tensor.expand torch.tensor.repeat torch.tensor.expand 先看招 import torch x = torch.tensor([ ...
通俗讲解Pytorch梯度的相关问题：计算图、torch.no_grad、zero_grad、detach和backward；Variable、Parameter和torch.tensor
文章目录 with torch.no_grad()和requires_grad backward() Variable,Parameter和torch.tensor() zero_grad() 计算图 ...
Unable to get repr for＜class‘torch.Tensor‘＞
Unable to get repr for <class 'torch.Tensor'> tensor越界访问后就会变成这样. import torcha_data=torch.Tens ...
unable to get repr for class ‘torch.tensor‘
unable to get repr for class 'torch.tensor' 出错代码: batch_conf.gather(1, conf_t.view(-1,1)) 最近码代码使用pyt ...
torch.Tensor和torch.tensor的区别
torch.Tensor和torch.tensor的区别 2019-06-10 16:34:48 Vic_Hao 阅读数 4058更多分类专栏: Pytorch 在Pytorch中,Tensor和t ...
c++ 打印 torch tensor
目录 c++ 打印 torch tensor cuda cu代码中打印: tensorflow::Tensor打印: 打印tensorrt cuda值: c++ 打印 torch tensor for ...
torch tensor用法
1.x = torch.Tensor(5):zero() Tensor转int: int(x1.cpu().data) range与arange区别: if __name__ == '__main__ ...
pytorch torch.Tensor.numpy()（从张量创建一个numpy数组，数组和张量共享相同内存）
https://pytorch.org/docs/1.1.0/tensors.html?highlight=numpy#torch.Tensor.numpy numpy() → numpy.ndarr ...

list转torch tensor

list转torch tensor相关推荐

最新文章

热门文章