Is there a way to pass an entire embedding to time_varying_unknown_reals without splitting it into individual components?

[LinearBasis]

Hello!

I have a question regarding the time_varying_unknown_reals parameter. Is there any way to pass an entire embedding (e.g., a dense vector representation) directly to time_varying_unknown_reals, rather than splitting it into its individual components?

For example, if I have an embedding of size n, can I pass it as a single feature, or do I need to split it into n separate features?

Thank you in advance for your help!