C/C++教程

Pytorch nn.Conv2d函数使用

本文主要是介绍Pytorch nn.Conv2d函数使用,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

Torch.nn.Conv2d(in_channels,out_channels,kernel_size,stride=1,padding=0,dilation=1,groups=1,bias=True)

in_channels:输入维度

out_channels:输出维度

kernel_size:卷积核大小

stride:步长大小

padding:补0

dilation:kernel间距

在这里插入图片描述

这篇关于Pytorch nn.Conv2d函数使用的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!