class GetInterpGradient : public GradientMakerBase{
using GradientMakerBase::GradientMakerBase;
vector<OperatorDef> GetGradientDefs() override{
return SingleGradientDef(
//梯度操作的type
"InterpGradient",
//梯度操作的name
"",
//梯度操作的输入
//GO():Output Gradient----前向计算的输出的梯度,即后向计算的输入
//I():前向计算的输入
//O():前向计算的输出
std::vector<string>{GO(0),I(0)},
//梯度操作的输出
//GI():Input Gradient----前向计算的输入的梯度,即后向计算的输出
std::vector<string>{GI{0}});
}
};
Caffe2填坑系列(4)----指定梯度计算时的输入和输出
最新推荐文章于 2023-01-04 14:16:00 发布