LibTorch之网络层中的卷积层

LibTorch之网络层中的卷积层

#include using namespace std;// 卷积层参数设置torch::nn::Conv2dOptions conv_options(int64_t in_planes, int64_t out_planes, int64_t kerner_size, int64_t stride = 1, int64_t padding = 0, bool with_bias = false) { torch::nn::Conv2dOptions conv_options = torch::nn::Conv2dOptions(in_planes, out_planes, kerner_size); conv_options.stride(stride); conv_options.padding(padding); conv_options.bias(with_bias); return conv_options;}// 卷积层torch::nn::Conv2d conv{ nullptr };int main(){ // 1.定义输入数据 torch::Tensor rgb = torch::randn({ 1, 3, 6, 6 }); cout << rgb << endl; cout << rgb.sizes() << endl; // 2.定义卷积层 conv = torch::nn::Conv2d(conv_options(3, 2, 5)); cout << "*********************" << endl; cout << "*********************" << endl; cout << "*********************" << endl; // 3.进行卷积 auto x = conv(rgb); // 4.卷积结果输出 cout << x << endl; cout << x.sizes() << endl; system("pause"); return 0;}

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。