netmap.model.nbautoencoder.get_mus¶

netmap.model.nbautoencoder.get_mus(model, data_tensor)[source]¶

Return the mean predicted mean expression (mu) across all cells.

Parameters:

model (NegativeBinomialAutoencoder) – A trained NB autoencoder.
data_tensor (torch.Tensor) – Input data on CPU; moved to CUDA internally.

Returns:

Mean mu value per gene, shape (input_dim,).

Return type: