β-VAE algorithm in the paper

[wxpbioinfo]

Hello, author,
In Figure 1b of your paper,I don't know why β-VAE can analyze the rate of cells of each cell type. I have studied this algorithm carefully and its input and output should correspond, so I don't understand why the input cell type is changed into the output of a single cell.
Could you please answer it, or what is the input data of this step?

[SpaTrek]

Thanks. The rate for each cell type is not calculated by the beta-vae. I think we have described the method in detail and included it in the paper. The rates are calculated first and beta-vae generates cells based on the rates.