Ejemplo n.º 1
0
 public override NDArray Backward(NDArray input, NDArray gradOutput, ModelMode mode)
 {
     SpatialConvolutionMM.Conv2BackwardInput(input, gradOutput, gradInput, weight, finput, fgradInput, cd);
     SpatialConvolutionMM.Conv2BackwardFilter(input, gradOutput, gradWeight, gradBias, finput, fgradInput, cd);
     return(gradInput);
 }
Ejemplo n.º 2
0
 public override Tensor Backward(Tensor input, Tensor gradOutput, ModelMode mode)
 {
     SpatialConvolutionMM.Conv2BackwardInput(input, gradOutput, gradInput, weight, finput, fgradInput, cd);
     SpatialConvolutionMM.Conv2BackwardFilter(input, gradOutput, gradWeight, gradBias, finput, fgradInput, cd);
     return(gradInput);
 }