Make output of transform same type as input

[jperez999]

This PR makes it so that the output to an executor transform is the same type as the input.

[github-actions[bot]]

Documentation preview

https://nvidia-merlin.github.io/core/review/pr-350