Browse Source

!6468 modify comments of vae

Merge pull request !6468 from byweng/master
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
656f9df5be
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      mindspore/nn/probability/dpn/vae/cvae.py
  2. +1
    -1
      mindspore/nn/probability/dpn/vae/vae.py

+ 1
- 1
mindspore/nn/probability/dpn/vae/cvae.py View File

@@ -43,7 +43,7 @@ class ConditionalVAE(Cell):
Inputs:
- **input_x** (Tensor) - The shape of input tensor is :math:`(N, C, H, W)`, which is the same as the input of
encoder.
encoder.
- **input_y** (Tensor) - The tensor of the target data, the shape is :math:`(N,)`.


+ 1
- 1
mindspore/nn/probability/dpn/vae/vae.py View File

@@ -40,7 +40,7 @@ class VAE(Cell):
Inputs:
- **input** (Tensor) - The shape of input tensor is :math:`(N, C, H, W)`, which is the same as the input of
encoder.
encoder.
Outputs:
- **output** (Tuple) - (recon_x(Tensor), x(Tensor), mu(Tensor), std(Tensor)).


Loading…
Cancel
Save