＜Pytorch,Libtorch 调试记录2＞Libtorch 深度学习之网络创建

云中飞鹤he

已于 2023-12-15 15:37:35 修改

阅读量121

点赞数

文章标签：深度学习网络人工智能

于 2023-11-17 14:23:08 首次发布

本文链接：https://blog.csdn.net/qq_35558042/article/details/134458272

版权

前言

使用PyTorch进行深度学习，创建网络非常的简单，但是为了更好的加速部署，最终落到了PyTorch的C++接口，也就是LibTorch，下文将具体介绍再C++中如何使用LibTorch进行深度神经网络的搭建。

方法

搭建一个三个卷积，两个全连接的网络进行分类计算
定义网络的结构体
1 声明
2 构造函数初始化
3 注册
4 前向传播

示例

#include <torch/torch.h>
#include <iostream>
#include <string>

struct Net:torch::nn::Module {
   
	//声明
	torch::nn::Conv2d conv1;
	torch::nn::Conv2d conv2;
	torch::nn::Conv2d conv3;
	torch::nn::Linear fc1;
	torch::nn::Linear fc2;

	//初始化
	Net() :
		conv1(torch::nn::Conv2dOptions(/*输入通道数*/