功能描述
算子功能
卷积算子。
算子公式
通用2D卷积计算公式:
参数 |
Depth |
Height |
Width |
Comment |
---|---|---|---|---|
Padding:Front,top and left |
PDL |
PHL |
PWL |
padding_l表示对应矢量的padding(l代表left)。 |
Padding:Back,bottom and right |
PDR |
PHR |
PWR |
padding_r表示对应矢量的padding(r代表right)。 |
Stride |
SD |
SH |
SW |
跳转距离,连续卷积的时候stride为1。 |
Dilation |
DD |
DH |
DW |
扩充值,无扩充时该值为0。 |
src |
- |
- |
- |
源输入Tensor。 |
weights |
- |
- |
- |
权重Tensor。 |
bias |
- |
- |
- |
偏置Tensor。 |
dst |
- |
- |
- |
输出结果Tensor。 |
ic |
- |
- |
- |
输入channel。 |
oc |
- |
- |
- |
输出channel。 |
oh |
- |
- |
- |
输出Height。 |
ow |
- |
- |
- |
输出Width。 |
kw |
- |
- |
- |
卷积核Width。 |
kh |
- |
- |
- |
卷积核Height。 |
父主题: Convolution