RandAffine does not follow the input tensor's device

[ziw-liu]

Describe the bug
A clear and concise description of what the bug is.

To Reproduce
Steps to reproduce the behavior:

1. Go to '...'
2. Install '....'
3. Run commands '....'

Expected behavior
A clear and concise description of what you expected to happen.

Screenshots
If applicable, add screenshots to help explain your problem.

Environment

Ensuring you use the relevant python executable, please paste the output of:

python -c "import monai; monai.config.print_debug_info()"

Additional context
Add any other context about the problem here.

[ziw-liu]

Couldn't reproduce on a different machine... Sorry for the noise!