!9575 Fix bug about grad of P.LinSpace.

From: @liu_xiao_93
Reviewed-by: @liangchenghui,@kingxian
Signed-off-by: @liangchenghui
pull/9575/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 6ff92a5b2e

@ -1191,7 +1191,7 @@ def get_bprop_inv(self):
def get_bprop_lin_space(self):
"""Grad definition for `LinSpace` operation."""
def bprop(assist, start, stop, num, out, dout):
return zeros_like(assist), zeros_like(start), zeros_like(stop), zeros_like(num)
def bprop(start, stop, num, out, dout):
return zeros_like(start), zeros_like(stop), zeros_like(num)
return bprop

Loading…
Cancel
Save