scglue.models.sc.NBDataDecoder.forward

NBDataDecoder.forward(u, v, b, l)[source]

Decode data from sample and feature latent

Parameters

  • u (Tensor) – Sample latent

  • v (Tensor) – Feature latent

  • b (Tensor) – Batch index

  • l (Tensor) – Optional normalizer

Return type

NegativeBinomial

Returns

recon – Data reconstruction distribution