Pytorch – torch.cat函数
1 torch.cat函数 形式 torch.cat(tensors, dim=0, *, out=None) 功能 在指定的维度连接给定序列的张量,所有张量必须具有相同的形状(连接维度除外)或者为空。 参数 tensors:相同形状的张量序列,非空张量必须具有相同形状(连接维度除外) dim:张量…
- Pytorch
- 2022-12-08
Pytorch – torch.unsqueeze和torch.squeeze函数
Pytorch中,unsqueeze和squeeze为两个对应的反操作函数,其中,unsqueeze主要用于为输入张量升维,squeeze主要用于给张量降维,两者的具体用法可以参考下文。 1 unsqueeze 形式 torch.unsqueeze(input, dim) 或者 Tensor.uns…
- Pytorch
- 2022-12-08
Pytorch – 创建随机Tensor的常用方法
本文对Pytorch中创建随机数Tensor的一些常用方法进行总结,主要包括 randn randn_like rand rand_like randint randint_like 1 torch.randn 形式 torch.randn(*size, *, out=None, dtype=Non…
- Pytorch
- 2022-12-08
Pytorch – torch.nn.Conv2d参数详解与使用
1 torch.nn.Conv2d torch.nn.Conv2d主要对输入Tensor应用2D卷积。 比如输入(N,C_{in},H,W)维度的Tensor,则输出(N,C_{out},H,W)的Tensor,这两者的关系可以描述为 \operatorname{out}\left(N_{i}, C…
- Pytorch
- 2022-12-07
TensorRT – onnx_graphsurgeon工具库的安装与API简介
1 onnx_grapgsurgeon 1.1 onnx_grapgsurgeon简介 onnx_grapgsurgeon是NVIDIA提供的创建和修改onnx模型的便捷易用工具库。 Github:https://github.com/NVIDIA/TensorRT/tree/master/tool…
- TensorRT
- 2022-12-05
TensorRT – 转换onnx模型出现Slice_74 requires bool or uint8 I/O but node can not be handled by Myelin错误
1 Error Code 2: Internal Error (Slice_74 requires bool or uint8 I/O but node can not be handled by Myelin错误 1.1 问题 今天在用TensorRT转化onnx模型出现了以下错误 转换onnx模…
- TensorRT
- 2022-12-02
Python – onnx导出模型出现RuntimeError: Exporting the operator pad_sequence to ONNX opset version 13 is not supported错误
1 onnx导出模型出现RuntimeError: Exporting the operator pad_sequence to ONNX opset version 13 is not supported错误 今天在pytorch中导出模型为onnx时,由于使用了from torch.nn.uti…
- 模型部署
- 2022-12-01
如何回看PLTV/m3u8直播源
1 回看m3u8直播源 一般来讲m3u8直播源不支持电视节目回看,不过如果远程服务器存储了过去的ts视频文件,那么该直播源就是支持回看的,一般的时间为过去7天,不过这都需要去试。 如果说一个m3u8直播源的链接为: http://39.134.66.66/PLTV/88888888/224/3221…
- 其他
- 2022-12-01
TensortRT – 转换模型出现Could not locate zlibwapi.dll. Please make sure it is in your library path!错误
1 TensortRT转换模型出现Could not locate zlibwapi.dll. Please make sure it is in your library path!错误 1.1 问题 今天在使用TensorRT转换onnx模型时出现了以下错误 [12/01/2022-09:03:…
- TensorRT
- 2022-12-01
Python – 使用onnxruntime加载和推理onnx模型
1 onnxruntime Onnx runtime是一个跨平台的机器学习模型加速器,可以在不同的硬件和操作系统上运行,可以加载和推理任意机器学习框架导出的onnx模型并进行加速。 如要使用onnxruntime,一般通过以下步骤: 从机器学习框架中将模型导出为onnx 使用onnxruntime加…
- 模型部署
- 2022-11-30
Python – 安装onnxruntime-gpu出现ERROR: Could not install packages due to an OSError: [Errno 2] No such file or directory: ‘…\\numpy-1.23.1.dist-info\\METADATA’
1 Python安装onnxruntime-gpu出错 今天在Anaconda中的虚拟环境中使用 pip install onnxruntime-gpu 安装onnxruntime的gpu版本库时出现了如下的错误 ERROR: Could not install packages due to an…
- Python
- 2022-11-29
C++ – 左值和右值,右值引用与移动语义的概念与理解
1 C++左值和右值 在看介绍C++11标准书籍的时候,经常在书中看到"左值"和"右值"的概念,这两个东西理解起来比较抽象。但是在C++11之后变得非常重要,其也是理解std::move()、std::forward等新语义的基础。 int a = 100; 上述示例代码中,a为左值,100为右值。…
- C++
- 2022-11-28
最新评论