fix error of WithBNNLossCell

289669a5 · bingyaweng · aed85287 · 289669a5 · 289669a5
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
@@ -13,6 +13,9 @@ mindspore.nn.probability.bnn_layers
 .. automodule:: mindspore.nn.probability.bnn_layers
    :members:
    
+    .. autoclass:: WithBNNLossCell
+        :members:
+    
 mindspore.nn.probability.distribution
 -------------------------------------


--- a/api/source_zh_cn/api/python/mindspore/mindspore.nn.probability.rst
+++ b/api/source_zh_cn/api/python/mindspore/mindspore.nn.probability.rst
@@ -13,6 +13,9 @@ mindspore.nn.probability.bnn_layers
 .. automodule:: mindspore.nn.probability.bnn_layers
    :members:
    
+    .. autoclass:: WithBNNLossCell    
+        :members:
+    
 mindspore.nn.probability.distribution
 -------------------------------------