From 5e63ecaa69eaf160bb00161a1899a2bc428544b2 Mon Sep 17 00:00:00 2001 From: bingyaweng Date: Fri, 18 Sep 2020 11:57:21 +0800 Subject: [PATCH] modify commit of vae --- mindspore/nn/probability/dpn/vae/cvae.py | 2 +- mindspore/nn/probability/dpn/vae/vae.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/nn/probability/dpn/vae/cvae.py b/mindspore/nn/probability/dpn/vae/cvae.py index 4996a1b7ef9..ee44326c0c8 100644 --- a/mindspore/nn/probability/dpn/vae/cvae.py +++ b/mindspore/nn/probability/dpn/vae/cvae.py @@ -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,)`. diff --git a/mindspore/nn/probability/dpn/vae/vae.py b/mindspore/nn/probability/dpn/vae/vae.py index 2993233f090..8fb7255657c 100644 --- a/mindspore/nn/probability/dpn/vae/vae.py +++ b/mindspore/nn/probability/dpn/vae/vae.py @@ -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)).