|
|
|
@ -50,7 +50,7 @@ class ControlDepend(Primitive):
|
|
|
|
|
>>> # step should be increased, so the add operation should depend on the data calculation operation.
|
|
|
|
|
>>> class Net(nn.Cell):
|
|
|
|
|
>>> def __init__(self):
|
|
|
|
|
>>> super(Net, self).__init__()
|
|
|
|
|
>>> super(Net, self).__init__()
|
|
|
|
|
>>> self.global_step = mindspore.Parameter(initializer(0, [1]), name="global_step")
|
|
|
|
|
>>> self.rate = 0.2
|
|
|
|
|
>>> self.control_depend = P.ControlDepend()
|
|
|
|
@ -89,7 +89,7 @@ class GeSwitch(PrimitiveWithInfer):
|
|
|
|
|
Examples:
|
|
|
|
|
>>> class Net(nn.Cell):
|
|
|
|
|
>>> def __init__(self):
|
|
|
|
|
>>> super(Net, self).__init__()
|
|
|
|
|
>>> super(Net, self).__init__()
|
|
|
|
|
>>> self.square = P.Square()
|
|
|
|
|
>>> self.add = P.TensorAdd()
|
|
|
|
|
>>> self.value = Tensor(np.full((1), 3), mindspore.float32)
|
|
|
|
|