diff --git a/mindspore/nn/layer/math.py b/mindspore/nn/layer/math.py index 765683526d3..506d8390b58 100644 --- a/mindspore/nn/layer/math.py +++ b/mindspore/nn/layer/math.py @@ -670,10 +670,10 @@ class IGamma(Cell): use_igammac = self.logicaland(self.greater(x, 1), self.greater(x, a)) ax = a * self.log(x) - x - self.lgamma(a) para_shape = self.shape(ax) - boradcastto = P.BroadcastTo(para_shape) + broadcastto = P.BroadcastTo(para_shape) if para_shape != (): - x = boradcastto(x) - y = boradcastto(y) + x = broadcastto(x) + a = broadcastto(a) x_is_zero = self.equal(x, 0) log_maxfloat = self.log_maxfloat32 underflow = self.less(ax, self.neg(log_maxfloat)) @@ -744,10 +744,10 @@ class LBeta(Cell): _check_input_dtype("input_y", y_dtype, x_dtype, self.cls_name) x_plus_y = x + y para_shape = self.shape(x_plus_y) - boradcastto = P.BroadcastTo(para_shape) + broadcastto = P.BroadcastTo(para_shape) if para_shape != (): - x = boradcastto(x) - y = boradcastto(y) + x = broadcastto(x) + y = broadcastto(y) comp_less = self.less(x, y) x_min = self.select(comp_less, x, y) y_max = self.select(comp_less, y, x)