revert

Author: sljlp

python/paddle/nn/layer/norm.py

@@ -1752,7 +1752,7 @@ class SpectralNorm(Layer):
 
     Step 1:
     Generate vector U in shape of [H], and V in shape of [W].
-    While H is the :attr:`axis` th dimension of the input weights,
+    While H is the :attr:`dim` th dimension of the input weights,
     and W is the product result of remaining dimensions.
 
     Step 2:
@@ -1779,9 +1779,9 @@ class SpectralNorm(Layer):
 
     Parameters:
         weight_shape(list or tuple): The shape of weight parameter.
-        axis(int, optional): The index of dimension which should be permuted to the first before reshaping Input(Weight) to matrix, it should be set as 0 if Input(Weight) is the weight of fc layer, and should be set as 1 if Input(Weight) is the weight of conv layer. Default: 0.
+        dim(int, optional): The index of dimension which should be permuted to the first before reshaping Input(Weight) to matrix, it should be set as 0 if Input(Weight) is the weight of fc layer, and should be set as 1 if Input(Weight) is the weight of conv layer. Default: 0.
         power_iters(int, optional): The number of power iterations to calculate spectral norm. Default: 1.
-        epsilon(float, optional): The epsilon for numerical stability in calculating norms. Default: 1e-12.
+        eps(float, optional): The epsilon for numerical stability in calculating norms. Default: 1e-12.
         name (str, optional): The default value is None.  Normally there is no need for user to set this property.  For more information, please refer to :ref:`api_guide_Name` .
         dtype (str, optional): Data type, it can be "float32" or "float64". Default: "float32".
 
@@ -1794,7 +1794,7 @@ class SpectralNorm(Layer):
             import paddle
             x = paddle.rand((2,8,32,32))
 
-            spectral_norm = paddle.nn.SpectralNorm(x.shape, axis=1, power_iters=2)
+            spectral_norm = paddle.nn.SpectralNorm(x.shape, dim=1, power_iters=2)
             spectral_norm_out = spectral_norm(x)
 
             print(spectral_norm_out.shape) # [2, 8, 32, 32]
@@ -1804,25 +1804,25 @@ class SpectralNorm(Layer):
     def __init__(
         self,
         weight_shape,
-        axis=0,
+        dim=0,
         power_iters=1,
-        epsilon=1e-12,
+        eps=1e-12,
         dtype='float32',
     ):
         super().__init__()
         self._power_iters = power_iters
-        self._epsilon = epsilon
-        self._dim = axis
+        self._epsilon = eps
+        self._dim = dim
         self._dtype = dtype
 
         self._weight_shape = list(weight_shape)
         assert (
             np.prod(self._weight_shape) > 0
         ), "Any dimension of `weight_shape` cannot be equal to 0."
-        assert axis < len(self._weight_shape), (
-            "The input `axis` should be less than the "
-            "length of `weight_shape`, but received axis="
-            "{}".format(axis)
+        assert dim < len(self._weight_shape), (
+            "The input `dim` should be less than the "
+            "length of `weight_shape`, but received dim="
+            "{}".format(dim)
         )
         h = self._weight_shape[self._dim]
         w = np.prod(self._weight_shape) // h