sccross.models.layers.NBDataDecoder.forward

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

Decode data from sample and feature latent

Parameters

  • u (Tensor) – Sample latent

  • v – Feature latent

  • b (Tensor) – Batch index

  • l (Tensor) – Optional normalizer

Return type

NegativeBinomial

Returns

recon – Data reconstruction distribution