How to context bake with Heatmap?

Hi,
I would like to use context bake Heatmap result with legend. It’s possible to context bake a geometry such as colored mesh but what about legend. May be it’s possible to transform a legend into a geometry ?
Thanks