Browse Source

!6680 update padding grad

Merge pull request !6680 from wuxuejian/padding_grad
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
190f0d84d3
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      mindspore/ops/_grad/grad_array_ops.py

+ 5
- 1
mindspore/ops/_grad/grad_array_ops.py View File

@@ -243,6 +243,10 @@ def get_bprop_embedding_lookup(self):
return RowTensor(new_indices, actual_dout, x_shp), zeros_like(indices), zeros_like(offset)
return bprop_sparse

@constexpr
def make_begin(shp):
begin = tuple([0 for _ in shp])
return begin

@bprop_getters.register(P.Padding)
def get_bprop_padding(self):
@@ -250,7 +254,7 @@ def get_bprop_padding(self):

def bprop(x, out, dout):
shp = shape_op(x)
begin = tuple([0 for _ in shp])
begin = make_begin(shp)
dx = P.Slice()(dout, begin, shp)
return (dx,)



Loading…
Cancel
Save