scvi.module.AmortizedLDAPyroModule.get_topic_distribution

AmortizedLDAPyroModule.get_topic_distribution(x, n_samples)[source]

Converts x to its inferred topic distribution.

Parameters
x : TensorTensor

Counts tensor.

n_samples : intint

Number of samples to take for the Monte-Carlo estimate of the mean.

Return type

TensorTensor

Returns

A x.shape[0] x n_topics tensor containing the normalized topic distribution.