Browse Source

!9575 Fix bug about grad of P.LinSpace.

From: @liu_xiao_93
Reviewed-by: @liangchenghui,@kingxian
Signed-off-by: @liangchenghui
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
6ff92a5b2e
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ops/_grad/grad_math_ops.py

+ 2
- 2
mindspore/ops/_grad/grad_math_ops.py View File

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