modify Probability api format

This commit is contained in:
huodagu 2022-06-10 08:30:45 +08:00
parent 15ec12a06b
commit a6d79af3ad
6 changed files with 8 additions and 8 deletions

View File

@ -34,8 +34,8 @@ mindspore.nn.probability.bnn_layers.ConvReparam
**输入:**
- **input** (Tensor) - 输入 Tensor 的 shape 为 :math:`(N, C_{in}`。
- **input** (Tensor) - 输入 Tensor 的 shape 为 :math:`(N, C_{in})`。
**输出:**
- **output** (Tensor) - 输出张量的形状是 :math:`(N, C_{out}, H_{out}, W_{out})`
Tensor输出张量的形状是 :math:`(N, C_{out}, H_{out}, W_{out})`

View File

@ -31,4 +31,4 @@ mindspore.nn.probability.bnn_layers.DenseLocalReparam
**输出:**
- **output** (Tensor) - `output` 的 shape 是 :math:`(N, out\_channels)`
Tensor`output` 的 shape 是 :math:`(N, out\_channels)`

View File

@ -33,4 +33,4 @@ mindspore.nn.probability.bnn_layers.DenseReparam
**输出:**
- **output** (Tensor) - `output` 的 shape 是 :math:`(N, out\_channels)`
Tensor`output` 的 shape 是 :math:`(N, out\_channels)`

View File

@ -1,7 +1,7 @@
mindspore.nn.probability.dpn.VAE
================================
.. py:class:: mindspore.nn.probability.dpn.VAE
.. py:class:: mindspore.nn.probability.dpn.VAE(encoder, decoder, hidden_size, latent_size)
变分自动编码器 (VAE)。
VAE 定义了一个生成模型Z从先验中采样然后由解码器用于重建X。有关更多详细信息请参阅 `自动编码变分贝叶斯 <https://arxiv.org/abs/1312.6114>`_
@ -36,7 +36,7 @@ mindspore.nn.probability.dpn.VAE
Tensor生成的样本。
.. py:method:: reconstruct_sample(×)
.. py:method:: reconstruct_sample(x)
从原始数据重建样本。

View File

@ -19,7 +19,7 @@ mindspore.nn.probability.infer.SVI
numpy.dtype训练后的损失。
.. py:method:: run(train_dataset, epochs =10)
.. py:method:: run(train_dataset, epochs=10)
通过训练概率网络来优化参数,并返回训练好的网络。

View File

@ -39,7 +39,7 @@ mindspore.nn.probability.toolbox.VAEAnomalyDetection
float样本的预测异常值分数。
.. py:method:: train(train_dataset , epochs =5)
.. py:method:: train(train_dataset , epochs=5)
训练 VAE 模型。