scvi.nn.Decoder.forward

Decoder.forward(x, *cat_list)[source]

The forward computation for a single sample.

  1. Decodes the data from the latent space using the decoder network

  2. Returns tensors for the mean and variance of a multivariate distribution

Parameters
x : TensorTensor

tensor with shape (n_input,)

cat_list : intint

list of category membership(s) for this sample

Returns

2-tuple of torch.Tensor Mean and variance tensors of shape (n_output,)