odels.vae_flax.FlaxDecoderOutput

<
source
>
(
sample: ndarray

)


Parameters 

sample (jnp.ndarray of shape (batch_size, num_channels, height, width)) —
Decoded output sample of the model. Output of the last layer of the model.


dtype (jnp.dtype, optional, defaults to jnp.float32) —
Parameters dtype



Output of decoding method.

replace

<
source
>
(
**updates

)



“Returns a new object replacing the specified fields with new values.

Flax