Two-axis billboarding within materials?

So I’ve figured out how to do one-axis billboarding, like for foliage, here:

It works by rotating around the Z-axis.

I would like to figure out how to **also **rotate around the Y-axis *or *X-axis.
I have a feeling it’s got to do with how the *custom *node is working, but I don’t know enough about HLSL to know how to change it.
What I want to achieve is something similar to how the ‘billboard’ component works in blueprints.

Anything? D:

Will that 3 vector change that? It does say “Rotates about the Z axis”. 0,0,1 = X,Y,Z?