pgjmblvch
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix TensorCopySlice bprop bug

This commit is contained in:
caifubi 2021-07-02 17:35:56 +08:00
parent d067d55f0f
commit c4914011e0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ops/_grad_experimental/grad_inner_ops.py
View File

@ -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)