Can you share a code snippet on what you are trying to do?
It depends on your model and use case.
Probably you have to train the network with both the variants of your data, if you want the behaviour to be same. .