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); }
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); }