Dear Sahil,
you can do that using the computeRho/ defineRho methods.
Background: for Advection-Diffusion equations, the zeroth moment is the scalar quantity (= the temperature in your case). This moment is often called “rho” in the LBM theory.
Yours,
Julius