![](https://img-blog.csdnimg.cn/20201014180756926.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
框架开发经验
lujingxi12
这个作者很懒,什么都没留下…
展开
-
c++ vector 元素互换
1111111111111111111111111111111111111111原创 2023-01-28 11:24:54 · 389 阅读 · 0 评论 -
注册工厂模式解释
需求:有若干个基类和它们的具体类,要设计一个统一的接口来实例化它们,且要用名字来取,方便管理。同一基类的子类注册到同一个工厂为什么要有注册模板类?不能用工厂直接注册吗?如果是这样,注册的内容是什么?是具体或抽象类吗?这样背离了通过一统一接口创建需要一个东西,把所有的类抽象一下,这就是注册模板类。2.为什么注册模板类需要有父类?不同的基类如果要注册到同一个工厂里,就要求工厂是模板工厂。注册...原创 2021-10-20 19:38:13 · 431 阅读 · 0 评论 -
在C++中如何手写onnx slice算子
slice 算子简介slice算子是对一个张量的某些轴进行切片获取数据。例如一个张量A的维度是 [d0, d1, d2,… di-1, di, di+1, … dn],如果对其第i轴进行切片(在此轴切取的数量为x),则操作后获得的张量维度为 [d0, d1, d2,… di-1, x, di+1, … dn]这种操作可以对多个轴进行操作,且每个轴可以指定starts index, ends index, steps(如果是2,则跳着2取数据,例如1,3,5,…)在python中,可以通过以下原创 2021-07-07 17:50:26 · 3415 阅读 · 0 评论 -
用python 接口写一个简单的ONNX卷积网络
用ONNX python 接口写一个简单的卷积网络本文提供初学者一个简单的案例,使用onnx python接口,构建一个输入shape为[1,1,3,3],输出为[1,3,3,3]的单一卷积网络。卷积的权重全为1,偏置全为2。构建ONNX模型的代码import onnx import numpy as npimport onnxruntimeinput = onnx.helper.make_tensor_value_info("input",onnx.TensorProto.FLOAT,[1,1原创 2021-04-06 13:42:39 · 544 阅读 · 0 评论 -
使用C++构建一个简单的卷积网络,并保存为ONNX模型
#include <iostream>#include <fstream>#include"onnx.pb.h"#include<string>using namespace std;int main(){onnx::GraphProto *graph = new onnx::GraphProto;graph->set_name("torch-jit-export");onnx::NodeProto* node1 = graph->add_原创 2021-04-06 19:38:35 · 622 阅读 · 0 评论