Question about applying moments to 'total' layer

[lijc0804]

Hello,

I am attempting to conduct an analysis based on smoothed expression by using the scv.pp.moments() function. However, I've noticed that only the 'spliced' and 'unspliced' layers are smoothed as 'Ms' and 'Mu'. But on my application, the 'total' is also provided in adata.layers, which means the total counts of each mRNA. Would it be possible to extend the moments processing to include the 'total' layer as well?

Thank you very much for providing such a marvelous tool for single-cell analysis!