| gradOutput |
输入 |
reflection_pad1d/reflection_pad2d的正向传播,shape支持2~4维且维度需要与self和gradInput保持一致,reflection_pad1d的反向算子对应维度为2/3维,reflection_pad2d的反向算子对应维度为3/4维。 |
FLOAT16、FLOAT32、DOUBLE、COMPLEX64、COMPLEX128 |
ND |
| self |
输入 |
shape支持2~4维且维度需要与self和gradInput保持一致,shape与gradInput一致。 |
FLOAT16、FLOAT32、DOUBLE、COMPLEX64、COMPLEX128 |
ND |
| padding |
输入 |
padding描述了向外填充的大小,长度为2或4。长度为2时表示1d左右需要填充的值,padding的前两个值都需要小于self最后一维度的数值。长度为4时表示2d左右上下需要填充的值,padding的前两个值都需要小于self最后一维度的数值,后两个值需要小于倒数第二维的数值。 |
INT64 |
|
| gradInput |
输出 |
数据类型与self保持一致。 |
FLOAT16、FLOAT32、DOUBLE、COMPLEX64、COMPLEX128 |
ND |