SetGradOutput
产品支持情况
产品 |
是否支持 |
|---|---|
√ |
|
√ |
|
x |
|
x |
|
x |
|
x |
|
x |
功能说明
设置卷积反向计算的输入矩阵GradOutput。
函数原型
1 | __aicore__ inline void SetGradOutput(const AscendC::GlobalTensor<SrcT> &gradOutput) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
gradOutput |
输入 |
GradOutput矩阵在Global Memory上的首地址。类型为GlobalTensor。SrcT表示GradOutput矩阵的数据类型，当前支持的数据类型为：half、bfloat16_t。 |
返回值说明
无
约束说明
无
调用示例
1 2 3 4 5 | ConvBackpropApi::Conv3DBackpropInput<weightDxType, inputSizeDxType, gradOutputDxType, gradInputDxType> gradInput_; // 设置GradOutput中GlobalTensor的地址 GlobalTensor<gradOutputType> gradOutputGm_; gradOutputGm_.SetGlobalBuffer((__gm__ gradOutputType *)gradOutput); gradInput_.SetGradOutput(gradOutputGm_); |
父主题： Conv3DBackpropInput