fix api param's issue.

4a8e972c · zhangyi · 2c8a97ba · 4a8e972c · 4a8e972c
2 changed file
--- a/api/source_en/api/python/mindspore/mindspore.nn.probability.rst
+++ b/api/source_en/api/python/mindspore/mindspore.nn.probability.rst
@@ -12,7 +12,14 @@ mindspore.nn.probability.bnn_layers

 .. automodule:: mindspore.nn.probability.bnn_layers
    :members:
+    :exclude-members: ConvReparam , DenseReparam
    
+    .. autoclass:: ConvReparam(in_channels, out_channels, kernel_size,stride=1, pad_mode='same', padding=0, dilation=1, group=1, has_bias=False, weight_prior_fn=NormalPrior, weight_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>, bias_prior_fn=NormalPrior, bias_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>)
+        :members:
+    
+    .. autoclass:: DenseReparam(in_channels, out_channels, activation=None, has_bias=True, weight_prior_fn=NormalPrior, weight_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>, bias_prior_fn=NormalPrior, bias_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>)
+        :members:
+            
    .. autoclass:: WithBNNLossCell
        :members:
    

--- a/api/source_zh_cn/api/python/mindspore/mindspore.nn.probability.rst
+++ b/api/source_zh_cn/api/python/mindspore/mindspore.nn.probability.rst
@@ -12,7 +12,14 @@ mindspore.nn.probability.bnn_layers

 .. automodule:: mindspore.nn.probability.bnn_layers
    :members:
+    :exclude-members: ConvReparam , DenseReparam
    
+    .. autoclass:: ConvReparam(in_channels, out_channels, kernel_size,stride=1, pad_mode='same', padding=0, dilation=1, group=1, has_bias=False, weight_prior_fn=NormalPrior, weight_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>, bias_prior_fn=NormalPrior, bias_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>)
+        :members:
+    
+    .. autoclass:: DenseReparam(in_channels, out_channels, activation=None, has_bias=True, weight_prior_fn=NormalPrior, weight_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>, bias_prior_fn=NormalPrior, bias_posterior_fn=<lambda name, shape: NormalPosterior(name=name, shape=shape)>)
+        :members:
+        
    .. autoclass:: WithBNNLossCell    
        :members: