From c4914011e07e1dac9040a240851b755d9f31a67f Mon Sep 17 00:00:00 2001 From: caifubi Date: Fri, 2 Jul 2021 17:35:56 +0800 Subject: [PATCH] fix TensorCopySlice bprop bug --- mindspore/ops/_grad_experimental/grad_inner_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ops/_grad_experimental/grad_inner_ops.py b/mindspore/ops/_grad_experimental/grad_inner_ops.py index 9ad74598b0f..47550a4a62f 100644 --- a/mindspore/ops/_grad_experimental/grad_inner_ops.py +++ b/mindspore/ops/_grad_experimental/grad_inner_ops.py @@ -25,7 +25,7 @@ def get_bprop_tensor_copy_slices(self): tensor_copy_slices = inner.TensorCopySlices() def bprop(x, update, begin, end, stride, out, dout): - x_grad = tensor_copy_slices(dout, F.zeros_like(update)) + x_grad = tensor_copy_slices(dout, F.zeros_like(update), begin, end, stride) update_grad = F.strided_slice(dout, begin, end, stride) return x_grad, update_grad, F.zeros_like(begin), F.zeros_like(end), F.zeros_like(stride)