forked from mindspore-Ecosystem/mindspore
modify commit of vae
This commit is contained in:
parent
fc7e7c9044
commit
5e63ecaa69
|
@ -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,)`.
|
||||
|
||||
|
|
|
@ -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…
Reference in New Issue